Searched refs:set_select (Results 1 – 4 of 4) sorted by relevance
209 instruction->set_select(new_select); in ReplaceComputations()
1861 void set_select(HloComputation* computation);
1369 void set_select(HloComputation* computation) { in set_select() function
4204 void HloInstruction::set_select(HloComputation* computation) { in set_select() function in xla::HloInstruction4205 return Cast<HloSelectAndScatterInstruction>(this)->set_select(computation); in set_select()