sourceShard (definition) in projects: llvm-project

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

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

/llvm-project/mlir/lib/Dialect/Mesh/Transforms/
H A D	Spmdization.cpp	61 TypedValue<ShapedType> sourceShard) { in handlePartialAxesDuringResharding() argument 134 TypedValue<ShapedType> sourceShard, MeshOp mesh, in splitLastAxisInResharding() argument 188 TypedValue<ShapedType> sourceShard) { in trySplitLastAxisInResharding() argument 262 unsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,MeshOp mesh,int64_t splitTensorAxis,MeshAxis splitMeshAxis) unsplitLastAxisInResharding() argument 288 tryUnsplitLastAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryUnsplitLastAxisInResharding() argument 393 moveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard,int64_t sourceTensorAxis,int64_t targetTensorAxis,MeshAxis meshAxis) moveLastSplitAxisInResharding() argument 421 tryMoveLastSplitAxisInResharding(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,ShapedType sourceUnshardedShape,TypedValue<ShapedType> sourceShard) tryMoveLastSplitAxisInResharding() argument 441 reshardOn1DMesh(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshardOn1DMesh() argument 484 reshard(ImplicitLocOpBuilder & builder,MeshOp mesh,MeshShardingAttr sourceSharding,MeshShardingAttr targetSharding,TypedValue<ShapedType> sourceUnshardedValue,TypedValue<ShapedType> sourceShard) reshard() argument [all...]
/llvm-project/mlir/test/lib/Dialect/Mesh/
H A D	TestReshardingSpmdization.cpp	`69 TypedValue<ShapedType> sourceShard = cast<TypedValue<ShapedType>>( matchAndRewrite() local`