Searched refs:set_while_body (Results 1 – 4 of 4) sorted by relevance
42 instruction->set_while_body(new_computation); in ReplaceCalledComputation()
156 instruction->set_while_body(new_body); in ReplaceComputations()
1067 void set_while_body(HloComputation* while_body);
2044 void HloInstruction::set_while_body(HloComputation* computation) { in set_while_body() function in xla::HloInstruction