Searched refs:set_called_computation (Results 1 – 2 of 2) sorted by relevance
1141 set_called_computation(kSelectComputationIndex, computation); in set_select()1148 set_called_computation(kScatterComputationIndex, computation); in set_scatter()
1647 void set_called_computation(int index, HloComputation* computation) { in set_called_computation() function