Lines Matching defs:yieldOperand
594 OpOperand *yieldOperand = getWarpResult(warpOp, [](Operation *op) {
597 if (!yieldOperand)
600 Operation *elementWise = yieldOperand->get().getDefiningOp();
601 unsigned operandIndex = yieldOperand->getOperandNumber();
660 OpOperand *yieldOperand =
662 if (!yieldOperand)
664 auto constantOp = yieldOperand->get().getDefiningOp<arith::ConstantOp>();
671 unsigned operandIndex = yieldOperand->getOperandNumber();
836 Value yieldOperand = yield.getOperand(result.getResultNumber());
838 std::make_pair(yieldOperand, newResultTypes.size()));
843 newYieldValues.push_back(yieldOperand);
1019 OpOperand *yieldOperand =
1021 if (!yieldOperand)
1024 auto mask = yieldOperand->get().getDefiningOp<vector::CreateMaskOp>();
1034 unsigned operandIndex = yieldOperand->getOperandNumber();
1586 for (OpOperand &yieldOperand : yield->getOpOperands()) {
1587 if (yieldOperand.get().getDefiningOp() != forOp.getOperation())
1589 auto forResult = cast<OpResult>(yieldOperand.get());
1591 newWarpOp.getResult(yieldOperand.getOperandNumber()));
1592 yieldOperand.set(forOp.getInitArgs()[forResult.getResultNumber()]);
1593 resultIdx.push_back(yieldOperand.getOperandNumber());
1689 OpOperand *yieldOperand =
1691 if (!yieldOperand)
1695 cast<vector::ReductionOp>(yieldOperand->get().getDefiningOp());
1712 unsigned operandIndex = yieldOperand->getOperandNumber();