Searched defs:targetSharding (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Spmdization.cpp | 60 MeshSharding targetSharding, in handlePartialAxesDuringResharding() argument
142 MeshSharding targetSharding = targetShardingInSplitLastAxis( in splitLastAxisInResharding() local
154 MeshSharding targetSharding) { in detectSplitLastAxisInResharding() argument
187 trySplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceShard) trySplitLastAxisInResharding() argument
204 detectUnsplitLastAxisInResharding(MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding) detectUnsplitLastAxisInResharding() argument
267 MeshShardingAttr targetSharding = unsplitLastAxisInResharding() local
286 tryUnsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryUnsplitLastAxisInResharding() argument
307 detectMoveLastSplitAxisInResharding(MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding) detectMoveLastSplitAxisInResharding() argument
399 MeshShardingAttr targetSharding = targetShardingInMoveLastAxis( moveLastSplitAxisInResharding() local
419 tryMoveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryMoveLastSplitAxisInResharding() argument
439 reshardOn1DMesh(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshardOn1DMesh() argument
482 reshard(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshard() argument
[all...] |