| /external/llvm-project/mlir/include/mlir/Dialect/SPIRV/ |
| D | SPIRVAttributes.h | 82 static VerCapExtAttr get(IntegerAttr version, ArrayAttr capabilities,
83 ArrayAttr extensions);
92 : public llvm::mapped_iterator<ArrayAttr::iterator,
94 explicit ext_iterator(ArrayAttr::iterator it);
101 ArrayAttr getExtensionsAttr();
104 : public llvm::mapped_iterator<ArrayAttr::iterator,
106 explicit cap_iterator(ArrayAttr::iterator it);
113 ArrayAttr getCapabilitiesAttr();
117 ArrayAttr capabilities,
118 ArrayAttr extensions);
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.h | 81 RankedTensorType filter_type, ArrayAttr strides, ArrayAttr dilations,
82 PatternRewriter& rewriter, ArrayAttr& explicit_pad);
85 ArrayAttr getPaddingValuesFromExplicitPadAttr(
86 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf,
94 ArrayAttr strides, ArrayAttr dilations, PatternRewriter& rewriter,
95 ArrayAttr& explicit_pad);
|
| D | legalize_utils.cc | 238 RankedTensorType filter_type, ArrayAttr strides, ArrayAttr dilations, in getPaddingValuesFromPadType()
239 PatternRewriter& rewriter, ArrayAttr& explicit_padding) { in getPaddingValuesFromPadType()
283 ArrayAttr getPaddingValuesFromExplicitPadAttr( in getPaddingValuesFromExplicitPadAttr()
284 ArrayAttr explicit_pad, tensorflow::TensorFormat data_format_tf, in getPaddingValuesFromExplicitPadAttr()
307 ArrayAttr strides, ArrayAttr dilations, PatternRewriter& rewriter, in getTransposeConv2dPaddingValues()
308 ArrayAttr& explicit_padding) { in getTransposeConv2dPaddingValues()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/ |
| D | SPIRVAttributes.cpp | 203 ArrayAttr capabilities, in get()
204 ArrayAttr extensions) { in get()
217 spirv::VerCapExtAttr::ext_iterator::ext_iterator(ArrayAttr::iterator it) in ext_iterator()
218 : llvm::mapped_iterator<ArrayAttr::iterator, in ext_iterator()
229 ArrayAttr spirv::VerCapExtAttr::getExtensionsAttr() { in getExtensionsAttr()
230 return getImpl()->extensions.cast<ArrayAttr>(); in getExtensionsAttr()
233 spirv::VerCapExtAttr::cap_iterator::cap_iterator(ArrayAttr::iterator it) in cap_iterator()
234 : llvm::mapped_iterator<ArrayAttr::iterator, in cap_iterator()
246 ArrayAttr spirv::VerCapExtAttr::getCapabilitiesAttr() { in getCapabilitiesAttr()
247 return getImpl()->capabilities.cast<ArrayAttr>(); in getCapabilitiesAttr()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| D | TosaOps.cpp | 151 Value bias, ArrayAttr pad, in buildConvOpWithQuantInfo()
152 ArrayAttr stride, ArrayAttr dilation) { in buildConvOpWithQuantInfo()
173 Value bias, ArrayAttr outpad, ArrayAttr stride, in buildTransConvOpWithQuantInfo()
174 ArrayAttr dilation, ArrayAttr outputShape) { in buildTransConvOpWithQuantInfo()
253 ArrayAttr kernel, ArrayAttr stride, in buildAvgPool2dOpWithQuantInfo()
254 ArrayAttr pad) { in buildAvgPool2dOpWithQuantInfo()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 27 inline bool isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul()
34 auto maps = ArrayAttr::get({mapA, mapB, mapC}, context); in isRowMajorMatmul()
38 inline bool isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul()
45 auto maps = ArrayAttr::get({mapA, mapB, mapC}, context); in isColumnMajorMatmul()
107 inline unsigned getNumIterators(StringRef name, ArrayAttr iteratorTypes) { in getNumIterators()
116 inline unsigned getNumIterators(ArrayAttr iteratorTypes) { in getNumIterators()
|
| /external/llvm-project/mlir/test/mlir-tblgen/ |
| D | predicate.td | 48 let arguments = (ins Confined<ArrayAttr, [ArrayMinCount<8>]>:$attr);
52 // CHECK: (tblgen_attr.cast<::mlir::ArrayAttr>().size() >= 8)
56 let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemEq<0, 8>]>:$attr);
60 // CHECK: (((tblgen_attr.cast<::mlir::ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<::mlir::Array…
64 let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemMinValue<0, 8>]>:$attr);
68 // CHECK: (((tblgen_attr.cast<::mlir::ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<::mlir::Array…
|
| /external/llvm-project/mlir/test/lib/Rewrite/ |
| D | TestPDLByteCode.cpp | 17 ArrayAttr constantParams, in customSingleEntityConstraint()
23 ArrayAttr constantParams, in customMultiEntityConstraint()
29 static PDLValue customCreate(ArrayRef<PDLValue> args, ArrayAttr constantParams, in customCreate()
37 ArrayAttr constantParams, in customRewriter()
|
| /external/llvm-project/mlir/test/mlir-linalg-ods-gen/ |
| D | test-linalg-ods-gen.tc | 10 // IMPL-LABEL: ArrayAttr Test1Op::iterator_types() {
13 // IMPL: ArrayAttr Test1Op::indexing_maps() {
35 // IMPL-LABEL: ArrayAttr Test2Op::iterator_types() {
38 // IMPL: ArrayAttr Test2Op::indexing_maps() {
60 // IMPL-LABEL: ArrayAttr Test3Op::iterator_types() {
63 // IMPL: ArrayAttr Test3Op::indexing_maps() {
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 37 class ArrayAttr; variable
97 ArrayAttr getArrayAttr(ArrayRef<Attribute> value);
135 ArrayAttr getAffineMapArrayAttr(ArrayRef<AffineMap> values);
136 ArrayAttr getBoolArrayAttr(ArrayRef<bool> values);
137 ArrayAttr getI32ArrayAttr(ArrayRef<int32_t> values);
138 ArrayAttr getI64ArrayAttr(ArrayRef<int64_t> values);
139 ArrayAttr getIndexArrayAttr(ArrayRef<int64_t> values);
140 ArrayAttr getF32ArrayAttr(ArrayRef<float> values);
141 ArrayAttr getF64ArrayAttr(ArrayRef<double> values);
142 ArrayAttr getStrArrayAttr(ArrayRef<StringRef> values);
[all …]
|
| D | PatternMatch.h | 312 ArrayRef<PDLValue>, ArrayAttr, PatternRewriter &)>;
317 std::function<PDLValue(ArrayRef<PDLValue>, ArrayAttr, PatternRewriter &)>;
322 ArrayAttr, PatternRewriter &)>;
328 std::function<LogicalResult(PDLValue, ArrayAttr, PatternRewriter &)>;
357 ArrayAttr, PatternRewriter &>::value>
361 ArrayRef<PDLValue> values, ArrayAttr constantParams, in registerConstraintFunction()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/ |
| D | TosaOpBase.td | 116 "ArrayAttr":$pad, "ArrayAttr":$stride, "ArrayAttr":$dilation),
126 "ArrayAttr":$outpad, "ArrayAttr":$stride, "ArrayAttr":$dilation,
127 "ArrayAttr":$outputShape),
159 (ins "Type":$outputType, "Value":$input, "ArrayAttr":$kernel,
160 "ArrayAttr":$stride, "ArrayAttr":$pad),
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 63 ArrayAttr createInt32Array(mlir::Builder* builder, mlir::MLIRContext* context, in createInt32Array()
69 return ArrayAttr::get(context, ret); in createInt32Array()
73 ArrayAttr createInt64Array(mlir::Builder* builder, mlir::MLIRContext* context, in createInt64Array()
79 return ArrayAttr::get(context, ret); in createInt64Array()
84 const ArrayAttr& pool_size, in createMaxUnpoolingAttr()
85 const ArrayAttr& strides) { in createMaxUnpoolingAttr()
|
| D | validators.cc | 30 auto attr = op->getAttrOfType<ArrayAttr>(name); in TFIntListIs1XY1()
51 bool TFIntListIs1XY1(const ArrayAttr &attr) { in TFIntListIs1XY1()
70 auto attr = op->getAttrOfType<ArrayAttr>(name); in TFIntListIs1XYZ1()
93 bool TFIntListIsAllOnes(const ArrayAttr &attr) { in TFIntListIsAllOnes()
|
| D | validators.h | 54 bool TFIntListIs1XY1(const ArrayAttr &attr);
65 bool TFIntListIsAllOnes(const ArrayAttr &attr);
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 202 ArrayAttr Builder::getArrayAttr(ArrayRef<Attribute> value) { in getArrayAttr()
203 return ArrayAttr::get(value, context); in getArrayAttr()
221 ArrayAttr Builder::getBoolArrayAttr(ArrayRef<bool> values) { in getBoolArrayAttr()
227 ArrayAttr Builder::getI32ArrayAttr(ArrayRef<int32_t> values) { in getI32ArrayAttr()
232 ArrayAttr Builder::getI64ArrayAttr(ArrayRef<int64_t> values) { in getI64ArrayAttr()
238 ArrayAttr Builder::getIndexArrayAttr(ArrayRef<int64_t> values) { in getIndexArrayAttr()
246 ArrayAttr Builder::getF32ArrayAttr(ArrayRef<float> values) { in getF32ArrayAttr()
252 ArrayAttr Builder::getF64ArrayAttr(ArrayRef<double> values) { in getF64ArrayAttr()
258 ArrayAttr Builder::getStrArrayAttr(ArrayRef<StringRef> values) { in getStrArrayAttr()
264 ArrayAttr Builder::getTypeArrayAttr(TypeRange values) { in getTypeArrayAttr()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | contraction_fusion.cc | 110 NamedAttribute fusion_attr(fusion_id, ArrayAttr::get(ctx, {fusion_name})); in matchAndRewrite()
114 ArrayAttr arr = in matchAndRewrite()
115 llvm::find_if(attrs, is_fusion)->second.template cast<ArrayAttr>(); in matchAndRewrite()
122 attrs.push_back(NamedAttribute(fusion_id, ArrayAttr::get(ctx, updated))); in matchAndRewrite()
|
| D | readonly_references_to_resources.cc | 66 ArrayAttr classes_attr = op->getAttrOfType<ArrayAttr>(kClassAttr); in GetNodeNameFromClassAttr()
72 classes_attr = identity_op->getAttrOfType<ArrayAttr>(kClassAttr); in GetNodeNameFromClassAttr()
|
| D | tpu_dynamic_padding_mapper.cc | 60 ArrayAttr replicated_input_indices_attr) { in GetRemappedReplicatedInputIndices()
100 auto padding_map = padding_map_attr.dyn_cast<ArrayAttr>(); in GetRemappedPaddings()
191 replicate->getAttrOfType<ArrayAttr>(kReplicatedInputIndicesAttr); in RemapAndAssignPaddingMaps()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | DropUnitDims.cpp | 99 static ArrayAttr replaceUnitDims(DenseSet<unsigned> &unitDims, in replaceUnitDims()
140 return ArrayAttr::get( in replaceUnitDims()
195 ArrayAttr iteratorTypes = op.iterator_types(); in matchAndRewrite()
208 ArrayAttr newIndexingMapAttr = in matchAndRewrite()
223 op.iterator_typesAttr(ArrayAttr::get(newIteratorTypes, context)); in matchAndRewrite()
233 ArrayAttr reassociation;
285 ArrayAttr::get(reassociationMaps, context)}; in replaceUnitExtents()
305 SmallVector<ArrayAttr, 4> reassociationMaps; in matchAndRewrite()
|
| D | Interchange.cpp | 80 ArrayAttr::get(newIndexingMaps, context)); in interchange()
82 ArrayAttr::get(itTypesVector, context)); in interchange()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 72 llvm::Optional<ArrayAttr> ExtractDilationsAttrFromBlockShape(
89 const ArrayAttr& dilations = in matchAndRewrite()
90 op->template getAttrOfType<ArrayAttr>("dilations"); in matchAndRewrite()
198 llvm::Optional<ArrayAttr> dilations_attr = ExtractDilationsAttrFromBlockShape( in matchAndRewrite()
308 llvm::Optional<ArrayAttr>
|
| /external/llvm-project/mlir/include/mlir/Interfaces/ |
| D | ViewLikeInterface.td | 118 /*retTy=*/"ArrayAttr",
130 /*retTy=*/"ArrayAttr",
142 /*retTy=*/"ArrayAttr",
252 static_offsets().template cast<ArrayAttr>(),
270 static_sizes().template cast<ArrayAttr>(), ShapedType::isDynamic, idx);
286 static_strides().template cast<ArrayAttr>(),
299 /*args=*/(ins "ArrayAttr":$attr,
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/ |
| D | RewriteInsertsPass.cpp | 80 auto indicesArrayAttr = op.indices().cast<ArrayAttr>(); in collectInsertionChain()
103 indicesArrayAttr = op.indices().cast<ArrayAttr>(); in collectInsertionChain()
|
| /external/llvm-project/debuginfo-tests/llvm-prettyprinters/gdb/ |
| D | mlir-support.cpp | 37 mlir::Attribute ArrayAttr = mlir::ArrayAttr::get({UnitAttr}, &Context); variable
|