| H A D | Spmdization.cpp | 59 MeshSharding sourceSharding, in handlePartialAxesDuringResharding() argument
110 targetShardingInSplitLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t splitTensorAxis,MeshAxis splitMeshAxis) targetShardingInSplitLastAxis() argument
133 splitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshShardingAttr sourceSharding,TypedValue<ShapedType> sourceShard,MeshOp mesh,int64_t splitTensorAxis,MeshAxis splitMeshAxis) splitLastAxisInResharding() argument
153 detectSplitLastAxisInResharding(MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding) detectSplitLastAxisInResharding() argument
186 trySplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceShard) trySplitLastAxisInResharding() argument
203 detectUnsplitLastAxisInResharding(MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding) detectUnsplitLastAxisInResharding() argument
233 targetShardingInUnsplitLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t splitTensorAxis) targetShardingInUnsplitLastAxis() argument
260 unsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,MeshOp mesh,int64_t splitTensorAxis,MeshAxis splitMeshAxis) unsplitLastAxisInResharding() argument
285 tryUnsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryUnsplitLastAxisInResharding() argument
306 detectMoveLastSplitAxisInResharding(MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding) detectMoveLastSplitAxisInResharding() argument
348 targetShardingInMoveLastAxis(MLIRContext * ctx,MeshShardingAttr sourceSharding,int64_t sourceTensorAxis,int64_t targetTensorAxis) targetShardingInMoveLastAxis() argument
391 moveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,int64_t sourceTensorAxis,int64_t targetTensorAxis,MeshAxis meshAxis) moveLastSplitAxisInResharding() argument
418 tryMoveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryMoveLastSplitAxisInResharding() argument
438 reshardOn1DMesh(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshardOn1DMesh() argument
481 reshard(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshard() argument
[all...] |