Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/Dialect/Mesh/Interfaces/
H A DShardingInterface.cpp626 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 DMeshShardingInterfaceImpl.cpp225 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