Searched refs:_test_collective_comms_gradients (Results 1 – 1 of 1) sorted by relevance
508 self._test_collective_comms_gradients(520 self._test_collective_comms_gradients(539 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, member in OneDeviceDistributionTestBase637 self._test_collective_comms_gradients(655 self._test_collective_comms_gradients(679 def _test_collective_comms_gradients(self, strategy, comm_fn, inputs, member in TwoDeviceDistributionTestBase