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