Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DSuperVectorize.cpp1300 auto it = strategy.reductionLoops.find(forOp); in vectorizeAffineForOp()
1301 assert(it != strategy.reductionLoops.end() && in vectorizeAffineForOp()
1659 const ReductionLoopMap &reductionLoops) { in vectorizeLoops() argument
1660 assert((reductionLoops.empty() || vectorSizes.size() == 1) && in vectorizeLoops()
1693 strategy.reductionLoops = reductionLoops; in vectorizeLoops()
1737 ReductionLoopMap reductionLoops; in runOnOperation() local
1742 f.walk([&parallelLoops, &reductionLoops](AffineForOp loop) { in runOnOperation()
1748 reductionLoops[loop] = reductions; in runOnOperation()
1761 reductionLoops); in runOnOperation()
1815 const ReductionLoopMap &reductionLoops) { in vectorizeAffineLoops() argument
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h96 ReductionLoopMap reductionLoops; member
116 const ReductionLoopMap &reductionLoops = ReductionLoopMap());