_test_all_reduce_sum_gradient_tape (reference) in projects: external

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

/external/tensorflow/tensorflow/python/distribute/
D	one_device_strategy_test.py	`105 self._test_all_reduce_sum_gradient_tape(distribution)`
D	strategy_test_lib.py	`528 def _test_all_reduce_sum_gradient_tape(self, strategy): member in OneDeviceDistributionTestBase 658 def _test_all_reduce_sum_gradient_tape(self, strategy, run_in_function=False): member in TwoDeviceDistributionTestBase`
D	mirrored_strategy_test.py	`244 self._test_all_reduce_sum_gradient_tape(distribution) 340 self._test_all_reduce_sum_gradient_tape(distribution)`
D	collective_all_reduce_strategy_test.py	`580 self._test_all_reduce_sum_gradient_tape(distribution)`
D	tpu_strategy_test.py	`1344 self._test_all_reduce_sum_gradient_tape(strategy, run_in_function=True)`