Lines Matching refs:ForOp
30 using scf::ForOp;
75 static void specializeForLoopForUnrolling(ForOp op) { in specializeForLoopForUnrolling()
111 static LogicalResult peelForLoop(RewriterBase &b, ForOp forOp, in peelForLoop()
112 ForOp &partialIteration, Value &splitBound) { in peelForLoop()
138 partialIteration = cast<ForOp>(b.clone(*forOp.getOperation())); in peelForLoop()
151 static void rewriteAffineOpAfterPeeling(RewriterBase &rewriter, ForOp forOp, in rewriteAffineOpAfterPeeling()
152 ForOp partialIteration, in rewriteAffineOpAfterPeeling()
176 ForOp forOp, in peelAndCanonicalizeForLoop()
177 ForOp &partialIteration) { in peelAndCanonicalizeForLoop()
196 struct ForLoopPeelingPattern : public OpRewritePattern<ForOp> {
198 : OpRewritePattern<ForOp>(ctx), skipPartial(skipPartial) {} in ForLoopPeelingPattern()
200 LogicalResult matchAndRewrite(ForOp forOp, in matchAndRewrite()
209 while ((op = op->getParentOfType<scf::ForOp>())) { in matchAndRewrite()
215 scf::ForOp partialIteration; in matchAndRewrite()
248 getOperation()->walk([](ForOp op) { specializeForLoopForUnrolling(op); }); in runOnOperation()