Home
last modified time | relevance | path

Searched refs:set_branch_computation (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dconditional_simplifier.cc128 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()
H A Dflatten_call_graph.cc57 instruction->set_branch_computation(b, new_computation); in ReplaceCalledComputation()
H A Dhlo_module.cc209 instruction->set_branch_computation(b, new_computation); in ReplaceComputations()
H A Dhlo_instruction.h1487 void set_branch_computation(int b, HloComputation* computation);
H A Dconditional_code_motion.cc1554 conditional->set_branch_computation(i, clone_i); in Run()
H A Dlayout_assignment.cc2449 caller->set_branch_computation( in Run()
H A Dhlo_instruction.cc2890 void HloInstruction::set_branch_computation(int b, in set_branch_computation() function in xla::HloInstruction