Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py108 self._test_all_reduce_sum_gradient_tape(distribution)
Dstrategy_test_lib.py511 def _test_all_reduce_sum_gradient_tape(self, strategy): member in OneDeviceDistributionTestBase
641 def _test_all_reduce_sum_gradient_tape(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
Dcollective_all_reduce_strategy_test.py493 self._test_all_reduce_sum_gradient_tape(distribution)
Dmirrored_strategy_test.py220 self._test_all_reduce_sum_gradient_tape(distribution)
316 self._test_all_reduce_sum_gradient_tape(distribution)
Dtpu_strategy_test.py1051 self._test_all_reduce_sum_gradient_tape(strategy, run_in_function=True)