Searched refs:_test_all_reduce_sum (Results 1 – 5 of 5) sorted by relevance
102 self._test_all_reduce_sum(distribution)
503 def _test_all_reduce_sum(self, strategy): member in OneDeviceDistributionTestBase628 def _test_all_reduce_sum(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
479 self._test_all_reduce_sum(distribution)
214 self._test_all_reduce_sum(distribution)310 self._test_all_reduce_sum(distribution)
1043 self._test_all_reduce_sum(strategy, run_in_function=True)