_test_all_reduce_mean_gradient_tape (reference) in projects: external

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

/external/tensorflow/tensorflow/python/distribute/
D	one_device_strategy_test.py	`114 self._test_all_reduce_mean_gradient_tape(distribution)`
D	strategy_test_lib.py	`540 def _test_all_reduce_mean_gradient_tape(self, strategy): member in OneDeviceDistributionTestBase 676 def _test_all_reduce_mean_gradient_tape(self, strategy, member in TwoDeviceDistributionTestBase`
D	mirrored_strategy_test.py	`253 self._test_all_reduce_mean_gradient_tape(distribution) 349 self._test_all_reduce_mean_gradient_tape(distribution)`
D	collective_all_reduce_strategy_test.py	`607 self._test_all_reduce_mean_gradient_tape(distribution)`
D	tpu_strategy_test.py	`1356 self._test_all_reduce_mean_gradient_tape(strategy, run_in_function=True)`