Lines Matching refs:yieldOperand

185       for (auto &yieldOperand : llvm::enumerate(yieldValues.getArrayRef())) {  in moveRegionToNewWarpOpAndAppendReturns()  local
186 if (yieldOperand.value() == std::get<0>(newRet)) { in moveRegionToNewWarpOpAndAppendReturns()
187 indices.push_back(yieldOperand.index()); in moveRegionToNewWarpOpAndAppendReturns()
214 for (OpOperand &yieldOperand : yield->getOpOperands()) { in getWarpResult()
215 Value yieldValues = yieldOperand.get(); in getWarpResult()
218 if (!warpOp.getResult(yieldOperand.getOperandNumber()).use_empty()) in getWarpResult()
219 return &yieldOperand; in getWarpResult()
481 OpOperand *yieldOperand = getWarpResult(warpOp, [](Operation *op) { in matchAndRewrite() local
484 if (!yieldOperand) in matchAndRewrite()
486 Operation *elementWise = yieldOperand->get().getDefiningOp(); in matchAndRewrite()
487 unsigned operandIndex = yieldOperand->getOperandNumber(); in matchAndRewrite()
545 OpOperand *yieldOperand = getWarpResult( in matchAndRewrite() local
547 if (!yieldOperand) in matchAndRewrite()
549 auto constantOp = yieldOperand->get().getDefiningOp<arith::ConstantOp>(); in matchAndRewrite()
553 unsigned operandIndex = yieldOperand->getOperandNumber(); in matchAndRewrite()
795 for (OpOperand &yieldOperand : yield->getOpOperands()) { in matchAndRewrite()
796 if (yieldOperand.get().getDefiningOp() != forOp.getOperation()) in matchAndRewrite()
798 auto forResult = yieldOperand.get().cast<OpResult>(); in matchAndRewrite()
799 newOperands.push_back(warpOp.getResult(yieldOperand.getOperandNumber())); in matchAndRewrite()
800 yieldOperand.set(forOp.getIterOperands()[forResult.getResultNumber()]); in matchAndRewrite()
801 resultIdx.push_back(yieldOperand.getOperandNumber()); in matchAndRewrite()
870 OpOperand *yieldOperand = getWarpResult( in matchAndRewrite() local
872 if (!yieldOperand) in matchAndRewrite()
876 cast<vector::ReductionOp>(yieldOperand->get().getDefiningOp()); in matchAndRewrite()
895 unsigned operandIndex = yieldOperand->getOperandNumber(); in matchAndRewrite()