Home
last modified time | relevance | path

Searched refs:isReduction (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorMultiDimReductionTransforms.cpp132 bool isReduction = it.value(); in matchAndRewrite() local
133 if (isReduction) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparsification.cpp1124 static bool isVectorFor(CodeGen &codegen, bool isInner, bool isReduction, in isVectorFor() argument
1127 if (codegen.sparseOut && !isReduction) in isVectorFor()
1144 static bool isParallelFor(CodeGen &codegen, bool isOuter, bool isReduction, in isParallelFor() argument
1154 return isOuter && !isSparse && !isReduction && !isVector; in isParallelFor()
1156 return isOuter && !isReduction && !isVector; in isParallelFor()
1158 return !isSparse && !isReduction && !isVector; in isParallelFor()
1160 return !isReduction && !isVector; in isParallelFor()
1198 bool isReduction = isReductionIterator(iteratorTypes[idx]); in genFor() local
1200 bool isVector = isVectorFor(codegen, isInner, isReduction, isSparse) && in genFor()
1203 isParallelFor(codegen, isOuter, isReduction, isSparse, isVector); in genFor()