Searched refs:_test_zero_join (Results 1 – 1 of 1) sorted by relevance
972 def _test_zero_join(self, device): member in TestZeroRedundancyOptimizerDistributed1125 self._test_zero_join(self.device)1131 self._test_zero_join(torch.device("cpu"))