Home
last modified time | relevance | path

Searched refs:ArrayAttr (Results 1 – 25 of 136) sorted by relevance

123456

/llvm-project-15.0.7/mlir/python/mlir/dialects/
H A D_structured_transform_ops_ext.py25 values: Optional[Union[ArrayAttr, Sequence[Attribute]]]) -> ArrayAttr: argument
28 return ArrayAttr.get([])
29 if isinstance(values, ArrayAttr):
32 return ArrayAttr.get(values)
37 ) -> ArrayAttr:
46 return ArrayAttr.get([])
47 if isinstance(values, ArrayAttr):
54 values: Optional[Union[ArrayAttr, Sequence[Union[ArrayAttr, argument
56 ) -> ArrayAttr:
65 return ArrayAttr.get([])
[all …]
H A D_ml_program_ops_ext.py96 return ArrayAttr(self.attributes[ARGUMENT_ATTRIBUTE_NAME])
99 def arg_attrs(self, attribute: Union[ArrayAttr, list]): argument
100 if isinstance(attribute, ArrayAttr):
103 self.attributes[ARGUMENT_ATTRIBUTE_NAME] = ArrayAttr.get(
115 def result_attrs(self, attribute: ArrayAttr): argument
H A D_pdl_ops_ext.py24 def _get_array_attr(attrs: Union[ArrayAttr, Sequence[Attribute]]) -> ArrayAttr: argument
26 if isinstance(attrs, ArrayAttr):
29 return ArrayAttr.get(list(attrs))
32 def _get_str_array_attr(attrs: Union[ArrayAttr, Sequence[str]]) -> ArrayAttr: argument
34 if isinstance(attrs, ArrayAttr):
37 return ArrayAttr.get([StringAttr.get(s) for s in attrs])
155 attributeNames = ArrayAttr.get(attributeNames)
H A D_func_ops_ext.py106 return ArrayAttr(self.attributes[ARGUMENT_ATTRIBUTE_NAME])
109 def arg_attrs(self, attribute: Union[ArrayAttr, list]): argument
110 if isinstance(attribute, ArrayAttr):
113 self.attributes[ARGUMENT_ATTRIBUTE_NAME] = ArrayAttr.get(
125 def result_attrs(self, attribute: ArrayAttr): argument
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DViewLikeInterface.h37 ArrayAttr staticOffsets,
43 ArrayAttr staticSizes,
49 ArrayAttr staticStrides,
55 std::pair<ArrayAttr, SmallVector<Value>> decomposeMixedStridesOrOffsets(
61 std::pair<ArrayAttr, SmallVector<Value>>
92 ArrayAttr integers);
105 OperandRange values, ArrayAttr integers);
123 ArrayAttr &integers);
141 ArrayAttr &integers);
146 Operation *op, StringRef name, unsigned expectedNumElements, ArrayAttr attr,
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SPIRV/IR/
H A DSPIRVAttributes.h90 static VerCapExtAttr get(IntegerAttr version, ArrayAttr capabilities,
91 ArrayAttr extensions);
100 : public llvm::mapped_iterator<ArrayAttr::iterator,
102 explicit ext_iterator(ArrayAttr::iterator it);
109 ArrayAttr getExtensionsAttr();
112 : public llvm::mapped_iterator<ArrayAttr::iterator,
114 explicit cap_iterator(ArrayAttr::iterator it);
121 ArrayAttr getCapabilitiesAttr();
125 ArrayAttr extensions);
157 ArrayAttr getExtensionsAttr();
[all …]
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DViewLikeInterface.cpp21 Operation *op, StringRef name, unsigned numElements, ArrayAttr attr, in verifyListOfOperandsOrIntegers()
75 ArrayAttr arrayAttr) { in printOperandsOrIntegersListImpl()
95 ArrayAttr integers) { in printOperandsOrIntegersOffsetsOrStridesList()
102 ArrayAttr integers) { in printOperandsOrIntegersSizesList()
111 ArrayAttr &integers) { in parseOperandsOrIntegersImpl()
148 ArrayAttr &integers) { in parseOperandsOrIntegersOffsetsOrStridesList()
156 ArrayAttr &integers) { in parseOperandsOrIntegersSizesList()
184 ArrayAttr staticOffsets, ValueRange offsets) { in getMixedOffsets()
214 ArrayAttr staticStrides, ValueRange strides) { in getMixedStrides()
227 static std::pair<ArrayAttr, SmallVector<Value>>
[all …]
/llvm-project-15.0.7/mlir/unittests/Dialect/Utils/
H A DStructuredOpsUtilsTest.cpp29 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
42 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
55 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
68 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
81 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
93 auto maps = ArrayAttr::get(&context, {mapA, mapB}); in TEST()
121 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
134 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
147 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
160 auto maps = ArrayAttr::get(&context, {mapA, mapB, mapC}); in TEST()
[all …]
/llvm-project-15.0.7/mlir/lib/IR/
H A DFunctionInterfaces.cpp29 ArrayAttr attrs = op->getAttrOfType<ArrayAttr>(getArgDictAttrName()); in getArgAttrDict()
38 ArrayAttr attrs = op->getAttrOfType<ArrayAttr>(getResultDictAttrName()); in getResultAttrDict()
47 ArrayAttr allAttrs = op->getAttrOfType<ArrayAttr>(attrName); in setArgResAttrDict()
56 op->setAttr(attrName, ArrayAttr::get(op->getContext(), newAttrs)); in setArgResAttrDict()
76 op->setAttr(attrName, ArrayAttr::get(op->getContext(), newAttrs)); in setArgResAttrDict()
85 op->setAttr(attrName, ArrayAttr::get(op->getContext(), attrs)); in setAllArgResAttrDicts()
131 auto oldArgAttrs = op->getAttrOfType<ArrayAttr>(getArgDictAttrName()); in insertFunctionArguments()
174 auto oldResultAttrs = op->getAttrOfType<ArrayAttr>(getResultDictAttrName()); in insertFunctionResults()
211 if (auto argAttrs = op->getAttrOfType<ArrayAttr>(getArgDictAttrName())) { in eraseFunctionArguments()
232 if (auto resAttrs = op->getAttrOfType<ArrayAttr>(getResultDictAttrName())) { in eraseFunctionResults()
[all …]
H A DBuilders.cpp205 ArrayAttr Builder::getArrayAttr(ArrayRef<Attribute> value) { in getArrayAttr()
206 return ArrayAttr::get(context, value); in getArrayAttr()
209 ArrayAttr Builder::getBoolArrayAttr(ArrayRef<bool> values) { in getBoolArrayAttr()
215 ArrayAttr Builder::getI32ArrayAttr(ArrayRef<int32_t> values) { in getI32ArrayAttr()
220 ArrayAttr Builder::getI64ArrayAttr(ArrayRef<int64_t> values) { in getI64ArrayAttr()
226 ArrayAttr Builder::getIndexArrayAttr(ArrayRef<int64_t> values) { in getIndexArrayAttr()
234 ArrayAttr Builder::getF32ArrayAttr(ArrayRef<float> values) { in getF32ArrayAttr()
240 ArrayAttr Builder::getF64ArrayAttr(ArrayRef<double> values) { in getF64ArrayAttr()
246 ArrayAttr Builder::getStrArrayAttr(ArrayRef<StringRef> values) { in getStrArrayAttr()
252 ArrayAttr Builder::getTypeArrayAttr(TypeRange values) { in getTypeArrayAttr()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Complex/IR/
H A DComplexOps.cpp31 if (auto arrAttr = value.dyn_cast<ArrayAttr>()) { in isBuildableWith()
43 ArrayAttr arrayAttr = getValue(); in verify()
85 ArrayAttr arrayAttr = operands[0].dyn_cast_or_null<ArrayAttr>(); in fold()
99 ArrayAttr arrayAttr = operands[0].dyn_cast_or_null<ArrayAttr>(); in fold()
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DStructuredOpsUtils.cpp15 bool mlir::isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul()
37 auto maps = ArrayAttr::get(context, {mapA, mapB, mapC}); in isRowMajorMatmul()
41 bool mlir::isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul()
63 auto maps = ArrayAttr::get(context, {mapA, mapB, mapC}); in isColumnMajorMatmul()
67 bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { in isRowMajorBatchMatmul()
90 auto maps = ArrayAttr::get(context, {mapA, mapB, mapC}); in isRowMajorBatchMatmul()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DStructuredOpsUtils.h34 bool isRowMajorMatmul(ArrayAttr indexingMaps);
40 bool isColumnMajorMatmul(ArrayAttr indexingMaps);
46 bool isRowMajorBatchMatmul(ArrayAttr indexingMaps);
109 inline unsigned getNumIterators(StringRef name, ArrayAttr iteratorTypes) { in getNumIterators()
118 inline unsigned getNumIterators(ArrayAttr iteratorTypes) { in getNumIterators()
188 ArrayAttr iterators;
H A DIndexingUtils.h22 class ArrayAttr; variable
47 SmallVector<int64_t, 4> getI64SubArray(ArrayAttr arrayAttr,
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVAttributes.cpp223 : llvm::mapped_iterator<ArrayAttr::iterator, in ext_iterator()
234 ArrayAttr spirv::VerCapExtAttr::getExtensionsAttr() { in getExtensionsAttr()
235 return getImpl()->extensions.cast<ArrayAttr>(); in getExtensionsAttr()
239 : llvm::mapped_iterator<ArrayAttr::iterator, in cap_iterator()
251 ArrayAttr spirv::VerCapExtAttr::getCapabilitiesAttr() { in getCapabilitiesAttr()
252 return getImpl()->capabilities.cast<ArrayAttr>(); in getCapabilitiesAttr()
258 ArrayAttr extensions) { in verify()
309 ArrayAttr spirv::TargetEnvAttr::getExtensionsAttr() { in getExtensionsAttr()
317 ArrayAttr spirv::TargetEnvAttr::getCapabilitiesAttr() { in getCapabilitiesAttr()
465 ArrayAttr capabilitiesAttr; in parseVerCapExtAttr()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferPermutationMapRewritePatterns.cpp25 static ArrayAttr
26 transposeInBoundsAttr(OpBuilder &builder, ArrayAttr attr, in transposeInBoundsAttr()
107 ArrayAttr newInBoundsAttr = in matchAndRewrite()
110 : ArrayAttr(); in matchAndRewrite()
177 ArrayAttr newInBoundsAttr = in matchAndRewrite()
180 : ArrayAttr(); in matchAndRewrite()
258 ArrayAttr newInBoundsAttr = in matchAndRewrite()
262 : ArrayAttr(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/test/python/ir/
H A Dattributes.py96 assert not ArrayAttr.isinstance(a1)
97 assert ArrayAttr.isinstance(a2)
457 for attr in ArrayAttr(raw):
464 raw = ArrayAttr.get([vecAttr, boolAttr, intAttr])
470 arr = ArrayAttr(raw)
486 ArrayAttr.get([None])
491 ArrayAttr.get([42])
497 array = ArrayAttr.get([StringAttr.get("a"), StringAttr.get("b")])
/llvm-project-15.0.7/mlir/test/mlir-tblgen/
H A Dpredicate.td69 let arguments = (ins Confined<ArrayAttr, [ArrayMinCount<8>]>:$attr);
73 // CHECK: (tblgen_attr.cast<::mlir::ArrayAttr>().size() >= 8)
77 let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemEq<0, 8>]>:$attr);
81 // CHECK: (((tblgen_attr.cast<::mlir::ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<::mlir::Array…
85 let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemMinValue<0, 8>]>:$attr);
89 // CHECK: (((tblgen_attr.cast<::mlir::ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<::mlir::Array…
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDropUnitDims.cpp99 static ArrayAttr replaceUnitDims(DenseSet<unsigned> &unitDims, in replaceUnitDims()
140 return ArrayAttr::get(context, in replaceUnitDims()
188 ArrayAttr iteratorTypes = genericOp.iterator_types(); in matchAndRewrite()
200 ArrayAttr newIndexingMapAttr = in matchAndRewrite()
215 genericOp.iterator_typesAttr(ArrayAttr::get(context, newIteratorTypes)); in matchAndRewrite()
225 ArrayAttr reassociation;
305 convertAffineMapArrayToExprs(ArrayAttr affineMapArrayAttr) { in convertAffineMapArrayToExprs()
320 ArrayAttr reassociationMap, Location loc, in maybeExpand()
340 ArrayAttr reassociationMap, Location loc, in maybeCollapse()
370 SmallVector<ArrayAttr> reassociationMaps; in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuilders.h38 class ArrayAttr; variable
105 ArrayAttr getArrayAttr(ArrayRef<Attribute> value);
140 ArrayAttr getAffineMapArrayAttr(ArrayRef<AffineMap> values);
141 ArrayAttr getBoolArrayAttr(ArrayRef<bool> values);
142 ArrayAttr getI32ArrayAttr(ArrayRef<int32_t> values);
143 ArrayAttr getI64ArrayAttr(ArrayRef<int64_t> values);
144 ArrayAttr getIndexArrayAttr(ArrayRef<int64_t> values);
145 ArrayAttr getF32ArrayAttr(ArrayRef<float> values);
146 ArrayAttr getF64ArrayAttr(ArrayRef<double> values);
147 ArrayAttr getStrArrayAttr(ArrayRef<StringRef> values);
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/IR/
H A DTosaOpBase.td130 "ArrayAttr":$pad, "ArrayAttr":$stride, "ArrayAttr":$dilation),
140 "ArrayAttr":$outpad, "ArrayAttr":$stride, "ArrayAttr":$outputShape),
172 (ins "Type":$outputType, "Value":$input, "ArrayAttr":$kernel,
173 "ArrayAttr":$stride, "ArrayAttr":$pad),
/llvm-project-15.0.7/mlir/test/lib/IR/
H A DTestFunc.cpp29 auto inserts = func->getAttrOfType<ArrayAttr>("test.insert_args"); in runOnOperation()
36 for (auto insert : inserts.getAsRange<ArrayAttr>()) { in runOnOperation()
67 auto inserts = func->getAttrOfType<ArrayAttr>("test.insert_results"); in runOnOperation()
73 for (auto insert : inserts.getAsRange<ArrayAttr>()) { in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp174 Value bias, ArrayAttr pad, in buildConvOpWithQuantInfo()
175 ArrayAttr stride, ArrayAttr dilation) { in buildConvOpWithQuantInfo()
197 ArrayAttr outpad, ArrayAttr stride, in buildTransConvOpWithQuantInfo()
198 ArrayAttr outputShape) { in buildTransConvOpWithQuantInfo()
274 ArrayAttr kernel, ArrayAttr stride, in buildAvgPool2dOpWithQuantInfo()
275 ArrayAttr pad) { in buildAvgPool2dOpWithQuantInfo()
543 ArrayAttr sizes = SliceOpAdaptor(operands, attributes).getSize(); in inferReturnTypeComponents()
575 ArrayAttr multiples = adaptor.getMultiples(); in inferReturnTypeComponents()
611 ArrayAttr newShape = adaptor.getNewShape(); in inferReturnTypeComponents()
995 getI64Values(attributes.get("kernel").cast<ArrayAttr>(), kernel); in NARY_SHAPE_INFER()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/EmitC/IR/
H A DEmitC.cpp87 if (Optional<ArrayAttr> argsAttr = getArgs()) { in verify()
97 } else if (arg.isa<ArrayAttr>() && arg.getType().isa<NoneType>()) { in verify()
103 if (Optional<ArrayAttr> templateArgsAttr = getTemplateArgs()) { in verify()
/llvm-project-15.0.7/mlir/test/python/dialects/
H A Dbuiltin.py187 f.arg_attrs = ArrayAttr.get([
192 DictAttr.get({"custom_dialect.qux": ArrayAttr.get([])})
194 f.result_attrs = ArrayAttr.get([

123456