Lines Matching defs:partialIteration

120                                  ForOp &partialIteration, Value &splitBound) {
157   partialIteration = cast<ForOp>(b.clone(*forOp.getOperation()));
158   partialIteration.getLowerBoundMutable().assign(splitBound);
159   b.replaceAllUsesWith(forOp.getResults(), partialIteration->getResults());
160   partialIteration.getInitArgsMutable().assign(forOp->getResults());
170                                         ForOp partialIteration,
173   Value partialIv = partialIteration.getInductionVar();
174   assert(forOp.getStep() == partialIteration.getStep() &&
186   partialIteration.walk([&](Operation *affineOp) {
197                                                       ForOp &partialIteration) {
200   if (failed(peelForLoop(rewriter, forOp, partialIteration, splitBound)))
204   rewriteAffineOpAfterPeeling(rewriter, forOp, partialIteration, previousUb);
264     scf::ForOp partialIteration;
268               peelForLoopFirstIteration(rewriter, forOp, partialIteration))) {
283               peelForLoopAndSimplifyBounds(rewriter, forOp, partialIteration)))
288     rewriter.modifyOpInPlace(partialIteration, [&]() {
289       partialIteration->setAttr(kPeeledLoopLabel, rewriter.getUnitAttr());
290       partialIteration->setAttr(kPartialIterationLabel, rewriter.getUnitAttr());