Searched defs:loopTypes (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 167 SmallVector<utils::IteratorType> loopTypes = getLoopIteratorTypes(); verifyShardingInterfaceImpl() local
261 SmallVector<utils::IteratorType> loopTypes = defaultGetShardingOption() local
394 getShardingAttribute(OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) getShardingAttribute() argument
474 SmallVector<utils::IteratorType> loopTypes = defaultGetShardingAnnotations() local
507 addShardOp(OpBuilder & b,OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) addShardOp() argument
538 SmallVector<utils::IteratorType> loopTypes = defaultAddShardingAnnotations() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 2823 build(OpBuilder & builder,OperationState & result,TypeRange loopTypes,Value target,ArrayRef<int64_t> staticTileSizes,ArrayRef<int64_t> interchange,std::optional<ArrayRef<bool>> scalableSizes) build() argument
2849 SmallVector<Type> loopTypes(1, builder.getType<transform::AnyOpType>()); build() local
2855 build(OpBuilder & builder,OperationState & result,TypeRange loopTypes,Value target,ArrayRef<OpFoldResult> mixedTileSizes,ArrayRef<int64_t> interchange,std::optional<ArrayRef<bool>> scalableSizes) build() argument
[all...] |