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