Searched defs:spmdizedOperands (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 171 createDestinationPassingStyleInitOperands(LinalgOp op,MeshOp meshOp,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshAxis> reductionMeshAxes,IRMapping & spmdizationMap,ImplicitLocOpBuilder & builder) createDestinationPassingStyleInitOperands() argument
221 spmdizeLinalgOpWithShardedReduction(LinalgOp op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,ImplicitLocOpBuilder & builder) spmdizeLinalgOpWithShardedReduction() argument
|
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 592 spmdizeFullyReplicatedOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,OpBuilder & builder) spmdizeFullyReplicatedOperation() argument
688 spmdizeTriviallyShardableOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTable,OpBuilder & builder) spmdizeTriviallyShardableOperation() argument
|
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Spmdization.cpp | 547 spmdizeOperation(Operation & op,ArrayRef<Value> spmdizedOperands,ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,IRMapping & spmdizationMap,SymbolTableCollection & symbolTableCollection,OpBuilder & builder) spmdizeOperation() argument
649 SmallVector<Value> spmdizedOperands; spmdizeOperation() local
|