Home
last modified time | relevance | path

Searched refs:arrayAttr (Results 1 – 20 of 20) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Complex/IR/
H A DComplexOps.cpp43 ArrayAttr arrayAttr = getValue(); in verify() local
44 if (arrayAttr.size() != 2) { in verify()
51 if (complexEltTy != arrayAttr[0].getType() || in verify()
52 complexEltTy != arrayAttr[1].getType()) { in verify()
55 << ", " << arrayAttr[1].getType() in verify()
85 ArrayAttr arrayAttr = operands[0].dyn_cast_or_null<ArrayAttr>(); in fold() local
86 if (arrayAttr && arrayAttr.size() == 2) in fold()
87 return arrayAttr[1]; in fold()
99 ArrayAttr arrayAttr = operands[0].dyn_cast_or_null<ArrayAttr>(); in fold() local
100 if (arrayAttr && arrayAttr.size() == 2) in fold()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationDialect.cpp88 auto arrayAttr = attr.getValue().dyn_cast<ArrayAttr>(); in verifyOperationAttribute() local
89 if (!arrayAttr) in verifyOperationAttribute()
92 if (arrayAttr.size() != op->getNumResults()) in verifyOperationAttribute()
96 << op->getNumResults() << ", got " << arrayAttr.size(); in verifyOperationAttribute()
101 for (const auto &it : llvm::enumerate(arrayAttr)) { in verifyOperationAttribute()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DIndexingUtils.cpp33 llvm::SmallVector<int64_t, 4> mlir::getI64SubArray(ArrayAttr arrayAttr, in getI64SubArray() argument
36 assert(arrayAttr.size() > dropFront + dropBack && "Out of bounds"); in getI64SubArray()
37 auto range = arrayAttr.getAsRange<IntegerAttr>(); in getI64SubArray()
39 res.reserve(arrayAttr.size() - dropFront - dropBack); in getI64SubArray()
H A DStaticValueUtils.cpp69 SmallVector<OpFoldResult> getAsOpFoldResult(ArrayAttr arrayAttr) { in getAsOpFoldResult() argument
71 res.reserve(arrayAttr.size()); in getAsOpFoldResult()
72 for (Attribute a : arrayAttr) in getAsOpFoldResult()
/llvm-project-15.0.7/mlir/test/lib/Dialect/Shape/
H A DTestShapeFunctions.cpp66 if (auto arrayAttr = attr.dyn_cast<ArrayAttr>()) { in runOnOperation() local
67 libraries.reserve(arrayAttr.size()); in runOnOperation()
68 for (auto attr : arrayAttr) in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Target/LLVMIR/
H A DModuleTranslation.cpp279 auto arrayAttr = attr.dyn_cast<ArrayAttr>(); in getLLVMConstant() local
280 if (!arrayAttr || arrayAttr.size() != 2) { in getLLVMConstant()
286 getLLVMConstant(elementType, arrayAttr[0], loc, moduleTranslation); in getLLVMConstant()
290 getLLVMConstant(elementType, arrayAttr[1], loc, moduleTranslation); in getLLVMConstant()
784 auto arrayAttr = attr.dyn_cast<ArrayAttr>(); in forwardPassthroughAttributes() local
785 if (!arrayAttr || arrayAttr.size() != 2) in forwardPassthroughAttributes()
789 auto keyAttr = arrayAttr[0].dyn_cast<StringAttr>(); in forwardPassthroughAttributes()
790 auto valueAttr = arrayAttr[1].dyn_cast<StringAttr>(); in forwardPassthroughAttributes()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DViewLikeInterface.cpp75 ArrayAttr arrayAttr) { in printOperandsOrIntegersListImpl() argument
77 if (arrayAttr.empty()) { in printOperandsOrIntegersListImpl()
82 llvm::interleaveComma(arrayAttr, p, [&](Attribute a) { in printOperandsOrIntegersListImpl()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp56 auto arrayAttr = attr.getValue().dyn_cast<ArrayAttr>(); in parse() local
57 if (!arrayAttr) { in parse()
62 for (auto i : arrayAttr) { in parse()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DIndexingUtils.h47 SmallVector<int64_t, 4> getI64SubArray(ArrayAttr arrayAttr,
H A DStaticValueUtils.h54 SmallVector<OpFoldResult> getAsOpFoldResult(ArrayAttr arrayAttr);
/llvm-project-15.0.7/mlir/test/lib/Dialect/Test/
H A DTestDialect.cpp1054 auto arrayAttr = getOperation()->getAttrOfType<ArrayAttr>("arg_names"); in getAsmBlockArgumentNames() local
1055 if (!arrayAttr) in getAsmBlockArgumentNames()
1058 auto e = std::min(arrayAttr.size(), args.size()); in getAsmBlockArgumentNames()
1060 if (auto strAttr = arrayAttr[i].dyn_cast<StringAttr>()) in getAsmBlockArgumentNames()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp330 static void getI64Values(ArrayAttr arrayAttr, SmallVector<int64_t> &values) { in getI64Values() argument
331 for (auto it : arrayAttr) { in getI64Values()
336 static void getF64Values(ArrayAttr arrayAttr, SmallVector<double> &values) { in getF64Values() argument
337 for (auto it : arrayAttr) { in getF64Values()
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DIRAttributes.cpp185 MlirAttribute arrayAttr = mlirArrayAttrGet( in bindDerived() local
187 return PyArrayAttribute(arr.getContext(), arrayAttr); in bindDerived()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp811 ArrayAttr arrayAttr = c.getValue(); in maybeZero() local
812 return arrayAttr[0].cast<FloatAttr>().getValue().isZero() && in maybeZero()
813 arrayAttr[0].cast<FloatAttr>().getValue().isZero(); in maybeZero()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp1069 static SmallVector<IntType> extractVector(ArrayAttr arrayAttr) { in extractVector() argument
1071 arrayAttr.getAsRange<IntegerAttr>(), in extractVector()
1591 static void populateFromInt64AttrArray(ArrayAttr arrayAttr, in populateFromInt64AttrArray() argument
1593 for (auto attr : arrayAttr) in populateFromInt64AttrArray()
2106 ArrayAttr arrayAttr, in isIntegerArrayAttrSmallerThanShape() argument
2109 if (arrayAttr.size() > shape.size()) in isIntegerArrayAttrSmallerThanShape()
2120 isIntegerArrayAttrConfinedToRange(OpType op, ArrayAttr arrayAttr, int64_t min, in isIntegerArrayAttrConfinedToRange() argument
2123 for (auto attr : arrayAttr) { in isIntegerArrayAttrConfinedToRange()
2140 isIntegerArrayAttrConfinedToShape(OpType op, ArrayAttr arrayAttr, in isIntegerArrayAttrConfinedToShape() argument
2143 assert(arrayAttr.size() <= shape.size()); in isIntegerArrayAttrConfinedToShape()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/IR/
H A DLLVMDialect.cpp2408 auto arrayAttr = getValue().dyn_cast<ArrayAttr>(); in verify() local
2409 if (!arrayAttr || arrayAttr.size() != 2 || in verify()
2410 arrayAttr[0].getType() != arrayAttr[1].getType()) { in verify()
/llvm-project-15.0.7/mlir/lib/Target/SPIRV/Serialization/
H A DSerializer.cpp664 } else if (auto arrayAttr = valueAttr.dyn_cast<ArrayAttr>()) { in prepareConstant() local
665 resultID = prepareArrayConstant(loc, constType, arrayAttr); in prepareConstant()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp141 static SmallVector<IntType, 4> extractVector(ArrayAttr arrayAttr) { in extractVector() argument
143 arrayAttr.getAsRange<IntegerAttr>(), in extractVector()
2232 static SmallVector<int64_t, 4> getIntValueVector(ArrayAttr arrayAttr) { in getIntValueVector() argument
2234 llvm::map_range(arrayAttr.getAsRange<IntegerAttr>(), in getIntValueVector()
/llvm-project-15.0.7/mlir/lib/IR/
H A DAsmPrinter.cpp1799 } else if (auto arrayAttr = attr.dyn_cast<ArrayAttr>()) { in printAttribute() local
1801 interleaveComma(arrayAttr.getValue(), [&](Attribute attr) { in printAttribute()
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVOps.cpp1865 if (auto arrayAttr = value.dyn_cast<ArrayAttr>()) { in verifyConstantType() local
1870 for (Attribute element : arrayAttr.getValue()) { in verifyConstantType()