Searched refs:_test_all_reduce_mean_gradients (Results 1 – 5 of 5) sorted by relevance
114 self._test_all_reduce_mean_gradients(distribution)
519 def _test_all_reduce_mean_gradients(self, strategy): member in OneDeviceDistributionTestBase654 def _test_all_reduce_mean_gradients(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
507 self._test_all_reduce_mean_gradients(distribution)
226 self._test_all_reduce_mean_gradients(distribution)322 self._test_all_reduce_mean_gradients(distribution)
1059 self._test_all_reduce_mean_gradients(strategy, run_in_function=True)