Searched refs:set_branch_computation (Results 1 – 7 of 7) sorted by relevance
126 conditional->set_branch_computation(branch, new_computation); in TryRemoveUnusedConditionalOperands()168 conditional_op->set_branch_computation(branch_id, branch_computation); in ReplaceRootWithEmptyTupleIfNoUsers()283 conditional_op->set_branch_computation(branch_id, cloned_branch); in RemoveUnusedTupleElements()
57 instruction->set_branch_computation(b, new_computation); in ReplaceCalledComputation()
203 instruction->set_branch_computation(b, new_computation); in ReplaceComputations()
1456 conditional->set_branch_computation(i, clone_i); in Run()
1419 void set_branch_computation(int b, HloComputation* computation);
2260 caller->set_branch_computation( in Run()
2729 void HloInstruction::set_branch_computation(int b, in set_branch_computation() function in xla::HloInstruction