Searched refs:_test_gloo_backend (Results 1 – 1 of 1) sorted by relevance
1498 def _test_gloo_backend( member in DistributedDataParallelTest1515 self._test_gloo_backend([torch.device("cpu")], None)1519 self._test_gloo_backend(1528 self._test_gloo_backend(devices, int_devices)1535 self._test_gloo_backend(devices, devices)1542 self._test_gloo_backend(devices, None, multi_device=True)1549 self._test_gloo_backend(devices, None, multi_device=True)