Searched defs:shardingOption (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 220 fillShardingOption(Operation * op,ShardingOption & shardingOption,FlatSymbolRefAttr mesh,ArrayRef<MeshAxis> meshAxes,unsigned loopIdx) fillShardingOption() argument
257 ShardingOption shardingOption; defaultGetShardingOption() local
393 getShardingAttribute(OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) getShardingAttribute() argument
435 getShardingAttribute(OpOperand & opOperand,const ShardingOption & shardingOption,AffineMap map) getShardingAttribute() argument
470 defaultGetShardingAnnotations(Operation * op,const ShardingOption & shardingOption) defaultGetShardingAnnotations() argument
505 addShardOp(OpBuilder & b,OpResult result,const ShardingOption & shardingOption,AffineMap map,ArrayRef<utils::IteratorType> loopTypes,ArrayRef<ReductionKind> reductionLoopKinds) addShardOp() argument
519 addShardOp(OpBuilder & b,OpOperand & opOperand,const ShardingOption & shardingOption,AffineMap map) addShardOp() argument
534 defaultAddShardingAnnotations(Operation * op,OpBuilder & b,const ShardingOption & shardingOption) defaultAddShardingAnnotations() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | ShardingPropagation.cpp | 230 FailureOr<ShardingOption> shardingOption = selectShardingOption() local
340 FailureOr<ShardingOption> shardingOption = selectShardingOption( visitOp() local
[all...] |