Lines Matching defs:rewriter

29 mlir::scf::forallToForLoop(RewriterBase &rewriter, scf::ForallOp forallOp,
31   OpBuilder::InsertionGuard guard(rewriter);
32   rewriter.setInsertionPoint(forallOp);
35   SmallVector<Value> lbs = forallOp.getLowerBound(rewriter);
36   SmallVector<Value> ubs = forallOp.getUpperBound(rewriter);
37   SmallVector<Value> steps = forallOp.getStep(rewriter);
38   LoopNest loopNest = scf::buildLoopNest(rewriter, loc, lbs, ubs, steps);
44   rewriter.eraseOp(forallOp.getBody()->getTerminator());
45   rewriter.inlineBlockBefore(forallOp.getBody(), innermostBlock,
48   rewriter.eraseOp(forallOp);
61     IRRewriter rewriter(parentOp->getContext());
64       if (failed(scf::forallToForLoop(rewriter, forallOp))) {