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