Searched defs:loopIteratorTypes (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Mesh/Interfaces/ |
| H A D | ShardingInterface.cpp | 623 getMeshAxisAssignmentForLoopIterators(ArrayRef<MeshShardingAttr> operandShardings,ArrayRef<MeshShardingAttr> resultShardings,ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<AffineMap> indexingMaps) getMeshAxisAssignmentForLoopIterators() argument
662 isAtLeastOneReductionIteratorSharded(ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators) isAtLeastOneReductionIteratorSharded() argument
675 getReductionMeshAxes(ArrayRef<utils::IteratorType> loopIteratorTypes,ArrayRef<SmallVector<MeshAxis>> meshAxisAssignmentForLoopIterators) getReductionMeshAxes() argument
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | MeshShardingInterfaceImpl.cpp | 224 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
314 SmallVector<utils::IteratorType> loopIteratorTypes = spmdize() local
|