Searched refs:called_computation_parameter (Results 1 – 1 of 1) sorted by relevance
1601 HloInstruction* called_computation_parameter = in AddCallOperand() local1605 return called_computation_parameter; in AddCallOperand()1661 HloInstruction* called_computation_parameter = in CloneAndAppendInstructionIntoCalledComputation() local1663 TF_CHECK_OK(called_computation_parameter->ReplaceAllUsesWith(clone)); in CloneAndAppendInstructionIntoCalledComputation()1697 HloInstruction* called_computation_parameter = nullptr; in CloneAndAppendInstructionIntoCalledComputation() local1700 called_computation_parameter = called_computation_parameters[i]; in CloneAndAppendInstructionIntoCalledComputation()1705 if (called_computation_parameter == nullptr) { in CloneAndAppendInstructionIntoCalledComputation()1709 called_computation_parameter = AddCallOperand(operand); in CloneAndAppendInstructionIntoCalledComputation()1712 clone->ReplaceOperandWith(operand_num, called_computation_parameter)); in CloneAndAppendInstructionIntoCalledComputation()