Searched defs:meshAxisAssignmentForLoopIterators (Results 1 – 2 of 2) sorted by relevance
/llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
H A D | ShardingInterface.cpp | 626 meshAxisAssignmentForLoopIterators(loopIteratorTypes.size()); getMeshAxisAssignmentForLoopIterators() local 663 isAtLeastOneReductionIteratorSharded(ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators) isAtLeastOneReductionIteratorSharded() argument 676 getReductionMeshAxes(ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators) getReductionMeshAxes() argument
|
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
H A D | MeshShardingInterfaceImpl.cpp | 225 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 316 ShardingArray meshAxisAssignmentForLoopIterators = spmdize() local
|