Lines Matching defs:operandIndex

601     unsigned operandIndex = yieldOperand->getOperandNumber();
602     Value distributedVal = warpOp.getResult(operandIndex);
635         {newWarpOp.getResult(operandIndex).getType()});
636     rewriter.replaceAllUsesWith(newWarpOp.getResult(operandIndex),
671     unsigned operandIndex = yieldOperand->getOperandNumber();
674         cast<ShapedType>(warpOp.getResult(operandIndex).getType()), scalarAttr);
678     rewriter.replaceAllUsesWith(warpOp.getResult(operandIndex), distConstant);
723     unsigned operandIndex = operand->getOperandNumber();
724     Value distributedVal = warpOp.getResult(operandIndex);
774     distributedVal = newWarpOp.getResult(operandIndex);
1034     unsigned operandIndex = yieldOperand->getOperandNumber();
1036     auto distType = cast<VectorType>(warpOp.getResult(operandIndex).getType());
1073     rewriter.replaceAllUsesWith(warpOp.getResult(operandIndex), newMask);
1712     unsigned operandIndex = yieldOperand->getOperandNumber();
1736     rewriter.replaceAllUsesWith(newWarpOp.getResult(operandIndex), fullReduce);