Searched defs:sharding (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/IR/ |
| H A D | MeshOps.cpp | 164 shardShapedType(ShapedType shape,MeshOp mesh,MeshShardingAttr sharding) shardShapedType() argument
172 shardType(Type type,MeshOp mesh,MeshShardingAttr sharding) shardType() argument
182 maybeInsertTargetShardingAnnotation(MeshShardingAttr sharding,OpOperand & operand,OpBuilder & builder) maybeInsertTargetShardingAnnotation() argument
214 maybeInsertTargetShardingAnnotation(MeshShardingAttr sharding,OpResult result,OpBuilder & builder) maybeInsertTargetShardingAnnotation() argument
222 maybeInsertSourceShardingAnnotation(MeshShardingAttr sharding,OpOperand & operand,OpBuilder & builder) maybeInsertSourceShardingAnnotation() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 109 if (sharding) { in getMesh() local
115 if (sharding) { in getMesh() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 566 isValueCompatibleWithFullReplicationSharding(Value value,MeshShardingAttr sharding) isValueCompatibleWithFullReplicationSharding() argument
[all...] |