Searched defs:indexingMaps (Results 1 – 8 of 8) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Generalization.cpp | 41 SmallVector<AffineMap, 4> indexingMaps = namedOp.getIndexingMaps(); in createGenericOpFromNamedOp() local
151 SmallVector<AffineMap, 4> indexingMaps = convOp.getIndexingMaps(); in createGenericOp() local
|
| D | Interchange.cpp | 66 auto indexingMaps = op.indexing_maps().getValue(); in interchange() local
|
| D | ElementwiseToLinalg.cpp | 39 SmallVector<AffineMap, 3> indexingMaps( in matchAndRewrite() local
|
| D | DropUnitDims.cpp | 100 ArrayRef<AffineMap> indexingMaps, in replaceUnitDims()
181 SmallVector<AffineMap, 4> indexingMaps = op.getIndexingMaps(); in matchAndRewrite() local
|
| D | Vectorization.cpp | 582 std::array<AffineMap, 3> indexingMaps{ in matchAndRewrite() local
|
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 27 inline bool isRowMajorMatmul(ArrayAttr indexingMaps) { in isRowMajorMatmul()
38 inline bool isColumnMajorMatmul(ArrayAttr indexingMaps) { in isColumnMajorMatmul()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 156 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
181 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
190 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
201 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
211 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
240 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
250 ValueRange outputBuffers, ArrayRef<AffineMap> indexingMaps, in build()
261 ArrayRef<AffineMap> indexingMaps, ArrayRef<StringRef> iteratorTypes, in build()
519 SmallVector<AffineMap, 4> indexingMaps; in verifyGenericOp() local
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 203 ArrayAttr indexingMaps, in build()
454 getDimMap(ArrayRef<AffineMap> indexingMaps, ArrayAttr iteratorTypes, in getDimMap()
475 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getIterationBounds() local
511 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getContractingDimMap() local
517 SmallVector<AffineMap, 4> indexingMaps(getIndexingMaps()); in getBatchDimMap() local
|