Searched refs:_test_all_reduce_mean (Results 1 – 5 of 5) sorted by relevance
111 self._test_all_reduce_mean(distribution)
515 def _test_all_reduce_mean(self, strategy): member in OneDeviceDistributionTestBase646 def _test_all_reduce_mean(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
500 self._test_all_reduce_mean(distribution)
223 self._test_all_reduce_mean(distribution)319 self._test_all_reduce_mean(distribution)
1055 self._test_all_reduce_mean(strategy, run_in_function=True)