Lines Matching refs:reductions
1298 ArrayRef<LoopReduction> reductions; in vectorizeAffineForOp() local
1303 reductions = it->second; in vectorizeAffineForOp()
1304 assert(reductions.size() == forOp.getNumIterOperands() && in vectorizeAffineForOp()
1317 for (auto redAndOperand : llvm::zip(reductions, forOp.getIterOperands())) { in vectorizeAffineForOp()
1357 getVectorReductionOp(reductions[i].kind, state.builder, in vectorizeAffineForOp()
1365 if (!isNeutralElementConst(reductions[i].kind, origInit, state)) in vectorizeAffineForOp()
1367 arith::getReductionOp(reductions[i].kind, state.builder, in vectorizeAffineForOp()
1743 SmallVector<LoopReduction, 2> reductions; in runOnOperation() local
1744 if (isLoopParallel(loop, &reductions)) { in runOnOperation()
1747 if (!reductions.empty()) in runOnOperation()
1748 reductionLoops[loop] = reductions; in runOnOperation()