| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 27 inline bool isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul() argument
28 auto context = indexingMaps.getContext(); in isRowMajorMatmul()
35 return indexingMaps == maps; in isRowMajorMatmul()
38 inline bool isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul() argument
39 auto context = indexingMaps.getContext(); in isColumnMajorMatmul()
46 return indexingMaps == maps; in isColumnMajorMatmul()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | DropUnitDims.cpp | 100 ArrayRef<AffineMap> indexingMaps, in replaceUnitDims() argument
102 if (indexingMaps.empty()) in replaceUnitDims()
104 unsigned numIterationDims = indexingMaps.front().getNumDims(); in replaceUnitDims()
105 unsigned numSymbols = indexingMaps.front().getNumSymbols(); in replaceUnitDims()
125 newIndexingMaps.reserve(indexingMaps.size()); in replaceUnitDims()
126 for (AffineMap operandMap : indexingMaps) { in replaceUnitDims()
181 SmallVector<AffineMap, 4> indexingMaps = op.getIndexingMaps(); in matchAndRewrite() local
182 if (indexingMaps.empty()) in matchAndRewrite()
188 AffineMap invertedMap = inversePermutation(concatAffineMaps(indexingMaps)); in matchAndRewrite()
209 replaceUnitDims(unitDims, indexingMaps, context); in matchAndRewrite()
|
| D | Generalization.cpp | 41 SmallVector<AffineMap, 4> indexingMaps = namedOp.getIndexingMaps(); in createGenericOpFromNamedOp() local
49 namedOp.getInitTensors(), indexingMaps, iterators, in createGenericOpFromNamedOp()
151 SmallVector<AffineMap, 4> indexingMaps = convOp.getIndexingMaps(); in createGenericOp() local
157 /*initTensors=*/ValueRange(), indexingMaps, iterators, in createGenericOp()
|
| D | Interchange.cpp | 66 auto indexingMaps = op.indexing_maps().getValue(); in interchange() local
68 AffineMap m = indexingMaps[i].cast<AffineMapAttr>().getValue(); in interchange()
|
| D | ElementwiseToLinalg.cpp | 39 SmallVector<AffineMap, 3> indexingMaps( in matchAndRewrite() local
49 /*indexingMaps=*/indexingMaps, in matchAndRewrite()
|
| D | Vectorization.cpp | 582 std::array<AffineMap, 3> indexingMaps{ in matchAndRewrite() local
591 rewriter.getAffineMapArrayAttr(indexingMaps), in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 156 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
160 builder.getAffineMapArrayAttr(indexingMaps), in build()
181 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build() argument
185 indexingMaps, iteratorTypes, doc, libraryCall, bodyBuild); in build()
190 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build() argument
193 build(builder, result, inputs, outputBuffers, indexingMaps, iteratorTypes, in build()
201 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
204 indexingMaps, iteratorTypes, in build()
211 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build() argument
216 builder.getAffineMapArrayAttr(indexingMaps), in build()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOps.td | 643 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
647 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
652 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
655 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
783 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
788 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
794 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
798 "ArrayRef<AffineMap>":$indexingMaps, "ArrayRef<StringRef>":$iteratorTypes,
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 203 ArrayAttr indexingMaps, in build() argument
207 result.addAttribute(getIndexingMapsAttrName(), indexingMaps); in build()
454 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap() argument
463 int64_t lhsDim = getResultIndex(indexingMaps[0], targetExpr); in getDimMap()
464 int64_t rhsDim = getResultIndex(indexingMaps[1], targetExpr); in getDimMap()
475 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getIterationBounds() local
483 int64_t lhsDimIndex = getResultIndex(indexingMaps[0], targetExpr); in getIterationBounds()
489 int64_t resDimIndex = getResultIndex(indexingMaps[2], targetExpr); in getIterationBounds()
511 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getContractingDimMap() local
512 return getDimMap(indexingMaps, iterator_types(), in getContractingDimMap()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorOps.td | 153 "ArrayAttr":$indexingMaps, "ArrayAttr":$iteratorTypes)>,
|