Lines Matching refs:yield
33 auto yield = cast<YieldOp>(headBlock->getTerminator()); in inlineIfCase() local
34 rewriter.setInsertionPoint(yield); in inlineIfCase()
35 rewriter.create<scf::YieldOp>(yield.getLoc(), yield.getInputs()); in inlineIfCase()
36 rewriter.eraseOp(yield); in inlineIfCase()
49 auto yield = cast<YieldOp>(headBlock->getTerminator()); in inlineWhileCase() local
50 rewriter.setInsertionPoint(yield); in inlineWhileCase()
53 rewriter.create<tensor::ExtractOp>(yield.getLoc(), yield.getOperand(0)); in inlineWhileCase()
54 rewriter.create<scf::ConditionOp>(yield.getLoc(), condition, in inlineWhileCase()
57 rewriter.setInsertionPoint(yield); in inlineWhileCase()
58 rewriter.create<scf::YieldOp>(yield.getLoc(), yield.getInputs()); in inlineWhileCase()
60 rewriter.eraseOp(yield); in inlineWhileCase()