Lines Matching refs:yieldOp

58     auto yieldOp = dyn_cast<scf::YieldOp>(  in getAliasingOpOperand()  local
60 assert(yieldOp && "expected scf.yield terminator in scf.execute_region"); in getAliasingOpOperand()
61 return {&yieldOp->getOpOperand(resultNum)}; in getAliasingOpOperand()
81 auto yieldOp = in bufferize() local
83 TypeRange newResultTypes(yieldOp.getResults()); in bufferize()
406 auto yieldOp = in bufferRelation() local
409 bbArg, yieldOp->getOperand(opResult.getResultNumber())); in bufferRelation()
441 auto yieldOp = in resolveConflicts() local
444 rewriter.setInsertionPoint(yieldOp); in resolveConflicts()
452 forOp.getRegionIterArgs(), yieldOp.getResults(), state); in resolveConflicts()
455 idx < static_cast<int64_t>(yieldOp.getResults().size()); ++idx) { in resolveConflicts()
456 Value value = yieldOp.getResults()[idx]; in resolveConflicts()
462 allocateTensorForShapedValue(rewriter, yieldOp.getLoc(), value, in resolveConflicts()
470 yieldOp, [&]() { yieldOp.getResultsMutable().assign(yieldValues); }); in resolveConflicts()
538 auto yieldOp = in verifyAnalysis() local
547 return yieldOp->emitError() in verifyAnalysis()
609 auto yieldOp = whileOp.getYieldOp(); in bufferRelation() local
611 Value yieldOperand = yieldOp.getOperand(resultNumber); in bufferRelation()
649 auto yieldOp = whileOp.getYieldOp(); in resolveConflicts() local
687 rewriter.setInsertionPoint(yieldOp); in resolveConflicts()
690 idx < static_cast<int64_t>(yieldOp.getResults().size()); ++idx) { in resolveConflicts()
691 Value value = yieldOp.getResults()[idx]; in resolveConflicts()
697 allocateTensorForShapedValue(rewriter, yieldOp.getLoc(), value, in resolveConflicts()
703 rewriter.updateRootInPlace(yieldOp, [&]() { in resolveConflicts()
704 yieldOp.getResultsMutable().assign(afterYieldValues); in resolveConflicts()
835 auto yieldOp = whileOp.getYieldOp(); in verifyAnalysis() local
836 for (const auto &it : llvm::enumerate(yieldOp.getResults())) { in verifyAnalysis()
840 it.value(), yieldOp->getBlock()->getArgument(it.index()))) in verifyAnalysis()
841 return yieldOp->emitError() in verifyAnalysis()
884 auto yieldOp = cast<scf::YieldOp>(op); in bufferize() local
886 yieldOp->getParentOp())) in bufferize()
887 return yieldOp->emitError("unsupported scf::YieldOp parent"); in bufferize()
891 if (isa<scf::WhileOp>(yieldOp->getParentOp())) in bufferize()
895 for (const auto &it : llvm::enumerate(yieldOp.getResults())) { in bufferize()
902 if (auto forOp = dyn_cast<scf::ForOp>(yieldOp->getParentOp())) { in bufferize()