Lines Matching refs:whileOp

575     auto whileOp = cast<scf::WhileOp>(op);  in getAliasingOpResult()  local
585 return {whileOp->getResult(idx)}; in getAliasingOpResult()
594 auto whileOp = cast<scf::WhileOp>(op); in bufferRelation() local
597 if (resultNumber >= whileOp.getBeforeArguments().size()) in bufferRelation()
600 whileOp.getBeforeArguments()[resultNumber].getType()) in bufferRelation()
603 auto conditionOp = whileOp.getConditionOp(); in bufferRelation()
604 BlockArgument conditionBbArg = whileOp.getBeforeArguments()[resultNumber]; in bufferRelation()
609 auto yieldOp = whileOp.getYieldOp(); in bufferRelation()
610 BlockArgument bodyBbArg = whileOp.getAfterArguments()[resultNumber]; in bufferRelation()
647 auto whileOp = cast<scf::WhileOp>(op); in resolveConflicts() local
648 auto conditionOp = whileOp.getConditionOp(); in resolveConflicts()
649 auto yieldOp = whileOp.getYieldOp(); in resolveConflicts()
653 DenseSet<int64_t> indicesBefore = getTensorIndices(whileOp.getInits()); in resolveConflicts()
655 getTensorIndices(whileOp.getAfterArguments()); in resolveConflicts()
660 whileOp.getBeforeArguments(), conditionOp.getArgs(), state); in resolveConflicts()
662 whileOp.getAfterArguments(), whileOp.getYieldOp().getResults(), state); in resolveConflicts()
714 auto whileOp = cast<scf::WhileOp>(op); in bufferize() local
716 assert(whileOp.getBefore().getBlocks().size() == 1 && in bufferize()
718 Block *beforeBody = &whileOp.getBefore().front(); in bufferize()
719 assert(whileOp.getAfter().getBlocks().size() == 1 && in bufferize()
721 Block *afterBody = &whileOp.getAfter().front(); in bufferize()
725 DenseSet<int64_t> indicesBefore = getTensorIndices(whileOp.getInits()); in bufferize()
727 getTensorIndices(whileOp.getAfterArguments()); in bufferize()
731 getBuffers(rewriter, whileOp->getOpOperands(), options); in bufferize()
738 llvm::map_range(whileOp.getAfterArguments(), [&](BlockArgument bbArg) { in bufferize()
746 auto newWhileOp = rewriter.create<scf::WhileOp>(whileOp.getLoc(), in bufferize()
750 SmallVector<Location> bbArgLocsBefore(initArgs.size(), whileOp.getLoc()); in bufferize()
752 whileOp.getLoc()); in bufferize()
818 auto whileOp = cast<scf::WhileOp>(op); in verifyAnalysis() local
824 auto conditionOp = whileOp.getConditionOp(); in verifyAnalysis()
835 auto yieldOp = whileOp.getYieldOp(); in verifyAnalysis()