Searched refs:_test_collective_comms_gradient_tape (Results 1 – 1 of 1) sorted by relevance
512 self._test_collective_comms_gradient_tape(524 self._test_collective_comms_gradient_tape(559 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, member in OneDeviceDistributionTestBase642 self._test_collective_comms_gradient_tape(661 self._test_collective_comms_gradient_tape(701 def _test_collective_comms_gradient_tape(self, strategy, comm_fn, inputs, member in TwoDeviceDistributionTestBase