Home
last modified time | relevance | path

Searched refs:iteratorType (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp515 for (const auto &iteratorType : enumerate(iteratorTypes)) in doit() local
516 if (isParallelIterator(iteratorType.value())) in doit()
517 parallelLoopRanges.push_back(loopRanges[iteratorType.index()]); in doit()
547 for (const auto &iteratorType : enumerate(iteratorTypes)) in doit() local
548 if (isParallelIterator(iteratorType.value())) in doit()
549 loops.push_back(loopNest.loops[iteratorType.index()]); in doit()
756 for (const auto &iteratorType : enumerate(iteratorTypes)) { in doit() local
757 if (isParallelIterator(iteratorType.value())) in doit()
765 for (const auto &iteratorType : enumerate(iteratorTypes)) { in doit() local
768 if (isParallelIterator(iteratorType.value())) { in doit()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp635 for (StringRef iteratorType : iteratorTypesRange) { in verifyStructuredOpInterface() local
636 if (!llvm::is_contained(getAllIteratorTypeNames(), iteratorType)) in verifyStructuredOpInterface()
638 << iteratorType << ")"; in verifyStructuredOpInterface()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp1019 for (const auto &iteratorType : llvm::enumerate(genericOp.iterator_types())) { in getCollapsableIterationSpaceDims() local
1020 if (isReductionIterator(iteratorType.value())) { in getCollapsableIterationSpaceDims()
1021 reductionDims.push_back(iteratorType.index()); in getCollapsableIterationSpaceDims()