Searched defs:sourceTensorAxis (Results 1 – 1 of 1) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Spmdization.cpp | 308 sourceTensorAxis < sourceSharding.getSplitAxes().size(); in detectMoveLastSplitAxisInResharding() local
349 targetShardingInMoveLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t sourceTensorAxis,int64_t targetTensorAxis) targetShardingInMoveLastAxis() argument
379 allToAllResultShapeInMoveLastAxis(ShapedType sourceShape,int64_t splitCount,int64_t sourceTensorAxis,int64_t targetTensorAxis) allToAllResultShapeInMoveLastAxis() argument
394 moveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,int64_t sourceTensorAxis,int64_t targetTensorAxis,MeshAxis meshAxis) moveLastSplitAxisInResharding() argument
[all...] |