Searched defs:splitTensorAxis (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Spmdization.cpp | 135 int64_t splitTensorAxis, MeshAxis splitMeshAxis) { in splitLastAxisInResharding() argument
111 targetShardingInSplitLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t splitTensorAxis,MeshAxis splitMeshAxis) targetShardingInSplitLastAxis() argument
234 targetShardingInUnsplitLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t splitTensorAxis) targetShardingInUnsplitLastAxis() argument
251 allGatherResultShapeInUnsplitLastAxis(ShapedType sourceShape,int64_t splitCount,int64_t splitTensorAxis) allGatherResultShapeInUnsplitLastAxis() argument
263 unsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,MeshOp mesh,int64_t splitTensorAxis,MeshAxis splitMeshAxis) unsplitLastAxisInResharding() argument
[all...] |