Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy_test.py102 self._test_all_reduce_sum(distribution)
Dstrategy_test_lib.py503 def _test_all_reduce_sum(self, strategy): member in OneDeviceDistributionTestBase
628 def _test_all_reduce_sum(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase
Dcollective_all_reduce_strategy_test.py479 self._test_all_reduce_sum(distribution)
Dmirrored_strategy_test.py214 self._test_all_reduce_sum(distribution)
310 self._test_all_reduce_sum(distribution)
Dtpu_strategy_test.py1043 self._test_all_reduce_sum(strategy, run_in_function=True)