Lines Matching refs:yieldOp

218       if (YieldOp yieldOp = dyn_cast<YieldOp>(blk.getTerminator())) {  in matchAndRewrite()  local
219 rewriter.setInsertionPoint(yieldOp); in matchAndRewrite()
220 rewriter.create<cf::BranchOp>(yieldOp.getLoc(), postBlock, in matchAndRewrite()
221 yieldOp.getResults()); in matchAndRewrite()
222 rewriter.eraseOp(yieldOp); in matchAndRewrite()
614 auto yieldOp = cast<scf::YieldOp>(block.getTerminator()); in matchAndRewrite() local
622 keepMask.reserve(yieldOp.getNumOperands()); in matchAndRewrite()
628 newYieldValues.reserve(yieldOp.getNumOperands()); in matchAndRewrite()
633 yieldOp.getOperands() // iter yield in matchAndRewrite()
768 auto yieldOp = cast<scf::YieldOp>(block.getTerminator()); in matchAndRewrite() local
769 auto yieldOperands = yieldOp.getOperands(); in matchAndRewrite()
976 auto yieldOp = in matchAndRewrite() local
978 Value yieldVal = yieldOp->getOperand(idx); in matchAndRewrite()
997 if (tensorLoadOp->getNextNode() != yieldOp) in matchAndRewrite()
1017 rewriter.startRootUpdate(yieldOp); in matchAndRewrite()
1018 yieldOp.setOperand(idx, bbArg); in matchAndRewrite()
1019 rewriter.finalizeRootUpdate(yieldOp); in matchAndRewrite()
1498 auto yieldOp = cast<scf::YieldOp>(dest->getTerminator()); in transferBody() local
1502 return yieldOp.getOperand(result.getResultNumber()); in transferBody()
1504 rewriter.updateRootInPlace(yieldOp, in transferBody()
1505 [&]() { yieldOp->setOperands(usedOperands); }); in transferBody()
2819 Operation *yieldOp = afterBlock.getTerminator(); in matchAndRewrite() local
2820 ValueRange yieldOpArgs = yieldOp->getOperands(); in matchAndRewrite()
2888 rewriter.setInsertionPoint(yieldOp); in matchAndRewrite()
2889 rewriter.replaceOpWithNewOp<YieldOp>(yieldOp, newYieldOpArgs); in matchAndRewrite()