Lines Matching refs:forOp
31 LogicalResult matchAndRewrite(ForOp forOp, in matchAndRewrite()
37 lcvTypes.push_back(forOp.getInductionVar().getType()); in matchAndRewrite()
38 lcvLocs.push_back(forOp.getInductionVar().getLoc()); in matchAndRewrite()
39 for (Value value : forOp.getInitArgs()) { in matchAndRewrite()
46 initArgs.push_back(forOp.getLowerBound()); in matchAndRewrite()
47 llvm::append_range(initArgs, forOp.getInitArgs()); in matchAndRewrite()
48 auto whileOp = rewriter.create<WhileOp>(forOp.getLoc(), lcvTypes, initArgs, in matchAndRewrite()
49 forOp->getAttrs()); in matchAndRewrite()
58 beforeBlock->getArgument(0), forOp.getUpperBound()); in matchAndRewrite()
71 whileOp.getLoc(), afterBlock->getArgument(0), forOp.getStep()); in matchAndRewrite()
75 for (const auto &barg : enumerate(forOp.getBody(0)->getArguments())) in matchAndRewrite()
79 for (auto &arg : llvm::make_early_inc_range(*forOp.getBody())) in matchAndRewrite()
93 for (const auto &arg : llvm::enumerate(forOp.getResults())) in matchAndRewrite()
96 rewriter.eraseOp(forOp); in matchAndRewrite()