Searched refs:set_branch_computation (Results 1 – 4 of 4) sorted by relevance
57 instruction->set_branch_computation(b, new_computation); in ReplaceCalledComputation()
165 instruction->set_branch_computation(b, new_computation); in ReplaceComputations()
1087 void set_branch_computation(int b, HloComputation* computation);
2087 void HloInstruction::set_branch_computation(int b, in set_branch_computation() function in xla::HloInstruction