Searched refs:set_branch_computation (Results 1 – 7 of 7) sorted by relevance
128 conditional->set_branch_computation(branch, new_computation); in TryRemoveUnusedConditionalOperands()170 conditional_op->set_branch_computation(branch_id, branch_computation); in ReplaceRootWithEmptyTupleIfNoUsers()285 conditional_op->set_branch_computation(branch_id, cloned_branch); in RemoveUnusedTupleElements()
57 instruction->set_branch_computation(b, new_computation); in ReplaceCalledComputation()
209 instruction->set_branch_computation(b, new_computation); in ReplaceComputations()
1487 void set_branch_computation(int b, HloComputation* computation);
1554 conditional->set_branch_computation(i, clone_i); in Run()
2449 caller->set_branch_computation( in Run()
2890 void HloInstruction::set_branch_computation(int b, in set_branch_computation() function in xla::HloInstruction