Searched refs:set_scatter (Results 1 – 4 of 4) sorted by relevance
179 instruction->set_scatter(new_scatter); in ReplaceComputations()
1578 void set_scatter(HloComputation* computation);
1144 void set_scatter(HloComputation* computation) { in set_scatter() function
3496 void HloInstruction::set_scatter(HloComputation* computation) { in set_scatter() function in xla::HloInstruction3497 return Cast<HloSelectAndScatterInstruction>(this)->set_scatter(computation); in set_scatter()