Searched refs:set_to_apply (Results 1 – 7 of 7) sorted by relevance
48 instruction->set_to_apply(new_computation); in ReplaceCalledComputation()
108 new_sort->set_to_apply(new_compare); in RemoveUnusedOperandFromSort()
93 sort->set_to_apply(sort->GetModule()->AddEmbeddedComputation( in ExpandInstruction()
290 root->set_to_apply(reduction_clone); in TEST_F()
143 instruction->set_to_apply(new_arg); in ReplaceComputations()
1058 void set_to_apply(HloComputation* to_apply);
2005 void HloInstruction::set_to_apply(HloComputation* computation) { in set_to_apply() function in xla::HloInstruction