Home
last modified time | relevance | path

Searched defs:targetTensorAxis (Results 1 – 1 of 1) sorted by relevance

/llvm-project/mlir/lib/Dialect/Mesh/Transforms/
H A DSpmdization.cpp311 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...]