Lines Matching defs:reductionLoops
1330 auto it = strategy.reductionLoops.find(forOp);
1331 assert(it != strategy.reductionLoops.end() &&
1686 /// vectorization order. `reductionLoops` can be provided to specify loops which
1691 const ReductionLoopMap &reductionLoops) {
1692 assert((reductionLoops.empty() || vectorSizes.size() == 1) &&
1725 strategy.reductionLoops = reductionLoops;
1766 ReductionLoopMap reductionLoops;
1769 // `reductionLoops` map.
1771 f.walk([¶llelLoops, &reductionLoops](AffineForOp loop) {
1777 reductionLoops[loop] = reductions;
1790 reductionLoops);
1837 /// If `reductionLoops` is not empty, the given reduction loops may be
1843 const ReductionLoopMap &reductionLoops) {
1847 reductionLoops);