Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_multiprocessing.py848 def _test_empty_tensor_sharing(self, dtype, device): member in TestMultiprocessing
856 self._test_empty_tensor_sharing(torch.float32, torch.device("cpu"))
857 self._test_empty_tensor_sharing(torch.int64, torch.device("cpu"))
861 self._test_empty_tensor_sharing(torch.float32, torch.device("cuda"))
862 self._test_empty_tensor_sharing(torch.int64, torch.device("cuda"))
865 self._test_empty_tensor_sharing(torch.float32, torch.device("meta"))
866 self._test_empty_tensor_sharing(torch.int64, torch.device("meta"))