Searched refs:_test_all_reduce_sum_gradient_tape (Results 1 – 5 of 5) sorted by relevance
108 self._test_all_reduce_sum_gradient_tape(distribution)
511 def _test_all_reduce_sum_gradient_tape(self, strategy): member in OneDeviceDistributionTestBase641 def _test_all_reduce_sum_gradient_tape(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
493 self._test_all_reduce_sum_gradient_tape(distribution)
220 self._test_all_reduce_sum_gradient_tape(distribution)316 self._test_all_reduce_sum_gradient_tape(distribution)
1051 self._test_all_reduce_sum_gradient_tape(strategy, run_in_function=True)