Searched refs:set_scatter (Results 1 – 4 of 4) sorted by relevance
223 instruction->set_scatter(new_scatter); in ReplaceComputations()
2080 void set_scatter(HloComputation* computation);
1765 void set_scatter(HloComputation* computation) { in set_scatter() function
4781 void HloInstruction::set_scatter(HloComputation* computation) { in set_scatter() function in xla::HloInstruction4782 return Cast<HloSelectAndScatterInstruction>(this)->set_scatter(computation); in set_scatter()