Home
last modified time | relevance | path

Searched refs:_test_all_reduce_mean_gradients (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py114 self._test_all_reduce_mean_gradients(distribution)
Dstrategy_test_lib.py519 def _test_all_reduce_mean_gradients(self, strategy): member in OneDeviceDistributionTestBase
654 def _test_all_reduce_mean_gradients(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
Dcollective_all_reduce_strategy_test.py507 self._test_all_reduce_mean_gradients(distribution)
Dmirrored_strategy_test.py226 self._test_all_reduce_mean_gradients(distribution)
322 self._test_all_reduce_mean_gradients(distribution)
Dtpu_strategy_test.py1059 self._test_all_reduce_mean_gradients(strategy, run_in_function=True)