Searched refs:bodyReturn (Results 1 – 1 of 1) sorted by relevance
61 auto bodyReturn = whileOp.body() in MatchAndRewrite() local69 if (!bodyReturn) return; in MatchAndRewrite()95 bodyReturn.getOperand(loopIndVar.second).getDefiningOp<mhlo::AddOp>(); in MatchAndRewrite()168 newYieldOps.reserve(bodyReturn.getNumOperands() - 1); in MatchAndRewrite()169 for (auto r : bodyReturn.getOperands().take_front(loopIndVar.second)) in MatchAndRewrite()171 for (auto r : bodyReturn.getOperands().drop_front(loopIndVar.second + 1)) in MatchAndRewrite()