Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py105 self._test_all_reduce_sum_gradients(distribution)
Dstrategy_test_lib.py507 def _test_all_reduce_sum_gradients(self, strategy): member in OneDeviceDistributionTestBase
636 def _test_all_reduce_sum_gradients(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
Dcollective_all_reduce_strategy_test.py486 self._test_all_reduce_sum_gradients(distribution)
Dmirrored_strategy_test.py217 self._test_all_reduce_sum_gradients(distribution)
313 self._test_all_reduce_sum_gradients(distribution)
Dtpu_strategy_test.py1047 self._test_all_reduce_sum_gradients(strategy, run_in_function=True)