Home
last modified time | relevance | path

Searched refs:isReductionIterator (Results 1 – 7 of 7) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/
H A DStructuredOpsUtils.h88 inline bool isReductionIterator(Attribute attr) { in isReductionIterator() function
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp298 isReductionIterator(iteratorTypes[2]))) in matchAndRewrite()
H A DVectorToGPU.cpp75 isReductionIterator(iteratorTypes[2]))) in contractSupportsMMAMatrixType()
297 isReductionIterator(iteratorTypes[2]))) in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp1020 if (isReductionIterator(iteratorType.value())) { in getCollapsableIterationSpaceDims()
1049 !isReductionIterator(startIteratorType)) in getCollapsableIterationSpaceDims()
1060 if (isReductionIterator(startIteratorType)) { in getCollapsableIterationSpaceDims()
H A DVectorization.cpp190 if (isReductionIterator(attr)) in getReductionMask()
537 if (llvm::none_of(op.iterator_types(), isReductionIterator)) { in reductionPreconditions()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp184 if (isReductionIterator(iteratorTypes[map.getDimPosition(i)])) in getReductionIndex()
1132 isReductionIterator(contractOp.getIteratorTypes() in matchAndRewrite()
1171 if (!isReductionIterator(iterators[i])) in matchAndRewrite()
1367 !isReductionIterator(iteratorTypes[2])) in matchAndRewrite()
1669 isReductionIterator(iteratorTypes[2])) { in matchAndRewrite()
1701 isReductionIterator(iteratorTypes[1])) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp355 if (isReductionIterator(iteratorTypes[topSort[i]])) in isAdmissableTensorExp()
1198 bool isReduction = isReductionIterator(iteratorTypes[idx]); in genFor()