Lines Matching refs:YieldOp
51 auto retValOp = dyn_cast<scf::YieldOp>(op); in handleTerminator()
76 builder.create<scf::YieldOp>(loc); in buildTerminatedBody()
218 if (YieldOp yieldOp = dyn_cast<YieldOp>(blk.getTerminator())) { in matchAndRewrite()
556 builder.create<scf::YieldOp>(loc, loops[i + 1].getResults()); in buildLoopNest()
570 builder.create<scf::YieldOp>(loc, results); in buildLoopNest()
614 auto yieldOp = cast<scf::YieldOp>(block.getTerminator()); in matchAndRewrite()
689 auto newYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in matchAndRewrite()
697 auto cloneFilteredTerminator = [&](scf::YieldOp mergedTerminator) { in matchAndRewrite()
705 rewriter.create<scf::YieldOp>(mergedTerminator.getLoc(), in matchAndRewrite()
710 auto mergedYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in matchAndRewrite()
768 auto yieldOp = cast<scf::YieldOp>(block.getTerminator()); in matchAndRewrite()
827 auto clonedYieldOp = cast<scf::YieldOp>(newBlock.getTerminator()); in replaceTensorCastForOpIterArg()
835 rewriter.create<scf::YieldOp>(newForOp.getLoc(), newYieldOperands); in replaceTensorCastForOpIterArg()
977 cast<scf::YieldOp>(forOp.getRegion().front().getTerminator()); in matchAndRewrite()
1498 auto yieldOp = cast<scf::YieldOp>(dest->getTerminator()); in transferBody()
1621 rewriter.replaceOpWithNewOp<YieldOp>(replacement.thenYield(), trueYields); in matchAndRewrite()
1624 rewriter.replaceOpWithNewOp<YieldOp>(replacement.elseYield(), falseYields); in matchAndRewrite()
1736 cast<scf::YieldOp>(op.getThenRegion().back().getTerminator()); in matchAndRewrite()
1738 cast<scf::YieldOp>(op.getElseRegion().back().getTerminator()); in matchAndRewrite()
1892 YieldOp thenYield = combinedIf.thenYield(); in matchAndRewrite()
1893 YieldOp thenYield2 = cast<YieldOp>(nextThen->getTerminator()); in matchAndRewrite()
1899 rewriter.create<YieldOp>(thenYield2.getLoc(), mergedYields); in matchAndRewrite()
1914 YieldOp elseYield = combinedIf.elseYield(); in matchAndRewrite()
1915 YieldOp elseYield2 = cast<YieldOp>(nextElse->getTerminator()); in matchAndRewrite()
1923 rewriter.create<YieldOp>(elseYield2.getLoc(), mergedElseYields); in matchAndRewrite()
2064 rewriter.replaceOpWithNewOp<YieldOp>(newIf.thenYield(), thenYield); in matchAndRewrite()
2068 rewriter.create<YieldOp>(loc, elseYield); in matchAndRewrite()
2086 YieldOp IfOp::thenYield() { return cast<YieldOp>(&thenBlock()->back()); } in thenYield()
2093 YieldOp IfOp::elseYield() { return cast<YieldOp>(&elseBlock()->back()); } in elseYield()
2564 YieldOp WhileOp::getYieldOp() { in getYieldOp()
2565 return cast<YieldOp>(getAfter().front().getTerminator()); in getYieldOp()
2705 auto afterTerminator = verifyAndGetTerminator<scf::YieldOp>( in verify()
2889 rewriter.replaceOpWithNewOp<YieldOp>(yieldOp, newYieldOpArgs); in matchAndRewrite()
3222 YieldOp yield = op.getYieldOp(); in matchAndRewrite()
3255 rewriter.replaceOpWithNewOp<YieldOp>(yield, newYields); in matchAndRewrite()