Lines Matching refs:yieldValues

177   llvm::SmallSetVector<Value, 32> yieldValues(yield.getOperands().begin(),  in moveRegionToNewWarpOpAndAppendReturns()  local
180 if (yieldValues.insert(std::get<0>(newRet))) { in moveRegionToNewWarpOpAndAppendReturns()
182 indices.push_back(yieldValues.size() - 1); in moveRegionToNewWarpOpAndAppendReturns()
185 for (auto &yieldOperand : llvm::enumerate(yieldValues.getArrayRef())) { in moveRegionToNewWarpOpAndAppendReturns()
193 yieldValues.insert(newYieldedValues.begin(), newYieldedValues.end()); in moveRegionToNewWarpOpAndAppendReturns()
195 rewriter, warpOp, yieldValues.getArrayRef(), types); in moveRegionToNewWarpOpAndAppendReturns()
215 Value yieldValues = yieldOperand.get(); in getWarpResult() local
216 Operation *definedOp = yieldValues.getDefiningOp(); in getWarpResult()
404 SmallVector<Value> yieldValues = {writeOp.getVector()}; in tryExtractOp() local
408 rewriter, warpOp, yieldValues, retTypes, newRetIndices); in tryExtractOp()
489 SmallVector<Value> yieldValues; in matchAndRewrite() local
506 yieldValues.push_back(operand.get()); in matchAndRewrite()
510 rewriter, warpOp, yieldValues, retTypes, newRetIndices); in matchAndRewrite()
631 SmallVector<Value> yieldValues; in matchAndRewrite() local
638 yieldValues.push_back(yield.getOperand(result.getResultNumber())); in matchAndRewrite()
640 if (yield.getNumOperands() == yieldValues.size()) in matchAndRewrite()
643 rewriter, warpOp, yieldValues, resultTypes); in matchAndRewrite()
662 SmallVector<Value> yieldValues; in matchAndRewrite() local
896 SmallVector<Value> yieldValues = {reductionOp.getVector()}; in matchAndRewrite() local
900 yieldValues.push_back(reductionOp.getAcc()); in matchAndRewrite()
905 rewriter, warpOp, yieldValues, retTypes, newRetIndices); in matchAndRewrite()