Lines Matching refs:partialIteration
112 ForOp &partialIteration, Value &splitBound) { in peelForLoop() argument
138 partialIteration = cast<ForOp>(b.clone(*forOp.getOperation())); in peelForLoop()
139 partialIteration.getLowerBoundMutable().assign(splitBound); in peelForLoop()
140 forOp.replaceAllUsesWith(partialIteration->getResults()); in peelForLoop()
141 partialIteration.getInitArgsMutable().assign(forOp->getResults()); in peelForLoop()
152 ForOp partialIteration, in rewriteAffineOpAfterPeeling() argument
155 Value partialIv = partialIteration.getInductionVar(); in rewriteAffineOpAfterPeeling()
156 assert(forOp.getStep() == partialIteration.getStep() && in rewriteAffineOpAfterPeeling()
167 partialIteration.walk([&](OpTy affineOp) { in rewriteAffineOpAfterPeeling()
177 ForOp &partialIteration) { in peelAndCanonicalizeForLoop() argument
180 if (failed(peelForLoop(rewriter, forOp, partialIteration, splitBound))) in peelAndCanonicalizeForLoop()
185 rewriter, forOp, partialIteration, previousUb); in peelAndCanonicalizeForLoop()
187 rewriter, forOp, partialIteration, previousUb); in peelAndCanonicalizeForLoop()
215 scf::ForOp partialIteration; in matchAndRewrite() local
216 if (failed(peelAndCanonicalizeForLoop(rewriter, forOp, partialIteration))) in matchAndRewrite()
219 partialIteration->setAttr(kPeeledLoopLabel, rewriter.getUnitAttr()); in matchAndRewrite()
223 partialIteration->setAttr(kPartialIterationLabel, rewriter.getUnitAttr()); in matchAndRewrite()