Searched refs:_test_collective_comms (Results 1 – 1 of 1) sorted by relevance
504 self._test_collective_comms(516 self._test_collective_comms(527 def _test_collective_comms(self, strategy, comm_fn, inputs, expected): member in OneDeviceDistributionTestBase629 self._test_collective_comms(647 self._test_collective_comms(665 def _test_collective_comms(self, strategy, comm_fn, inputs, expected, member in TwoDeviceDistributionTestBase