| /llvm-project/mlir/include/mlir/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterfaceImpl.h | 60 getLoopIteratorTypes(Operation *operation) const { in getLoopIteratorTypes()
109 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getLoopIteratorTypes()
58 getLoopIteratorTypes(Operation *operation) const { getLoopIteratorTypes() function
107 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { getLoopIteratorTypes() function
|
| H A D | ShardingInterface.td | 34 /*methodName=*/"getLoopIteratorTypes",
43 `getLoopIteratorTypes`.
|
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 178 SmallVector<utils::IteratorType> loopTypes = getLoopIteratorTypes(); in verifyShardingInterfaceImpl()
213 for (utils::IteratorType type : getLoopIteratorTypes()) {
274 shardingOp.getLoopIteratorTypes(); in defaultGetShardingOption()
489 shardingOp.getLoopIteratorTypes(); in defaultGetShardingAnnotations()
553 shardingOp.getLoopIteratorTypes(); in defaultAddShardingAnnotations()
|
| /llvm-project/mlir/lib/Dialect/Tosa/IR/ |
| H A D | ShardingInterfaceImpl.cpp | 34 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getLoopIteratorTypes() function
|
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 30 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getLoopIteratorTypes()
123 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getTiledImplementation()
496 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getIterationDomainTileFromOperandTile()
29 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { getLoopIteratorTypes() function
105 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { getLoopIteratorTypes() function
358 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { getLoopIteratorTypes() function
|
| /llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgOps.td | 101 "getLoopIteratorTypes",
161 "getLoopIteratorTypes",
232 "getLoopIteratorTypes",
321 "getLoopIteratorTypes",
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 271 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getIndexingMaps()
264 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { getLoopIteratorTypes() function
|
| H A D | TilingInterfaceImpl.cpp | 90 SmallVector<utils::IteratorType> getLoopIteratorTypes(Operation *op) const { in getLoopIteratorTypes() function
|
| H A D | Tiling.cpp | 626 tilingInterfaceOp.getLoopIteratorTypes(); in tileLinalgOpImpl()
|
| /llvm-project/mlir/include/mlir/Interfaces/ |
| H A D | TilingInterface.td | 46 - `getLoopIteratorTypes`
70 /*methodName=*/"getLoopIteratorTypes",
|
| /llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 162 auto iterators = op.getLoopIteratorTypes(); in generateLoopNestUsingForOp()
598 llvm::enumerate(op.getLoopIteratorTypes())) { in tileUsingSCF()
633 llvm::enumerate(op.getLoopIteratorTypes())) { in tileUsingSCF()
671 llvm::enumerate(op.getLoopIteratorTypes())) { in tileUsingSCF()
707 llvm::enumerate(op.getLoopIteratorTypes())) { in tileReductionUsingScf()
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 3017 if (tileSizes.size() > tilingInterface.getLoopIteratorTypes().size()) { in apply()
3021 << tilingInterface.getLoopIteratorTypes().size() << " found " in apply()
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 2641 SmallVector<utils::IteratorType> SoftmaxOp::getLoopIteratorTypes() { in buildSubAndExpOp()
2966 WinogradFilterTransformOp::getLoopIteratorTypes() {
3113 WinogradInputTransformOp::getLoopIteratorTypes() {
3290 WinogradOutputTransformOp::getLoopIteratorTypes() {
2478 SmallVector<utils::IteratorType> SoftmaxOp::getLoopIteratorTypes() { getLoopIteratorTypes() function in SoftmaxOp
|