Searched refs:_spawn_method (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cuda.py | 954 def _spawn_method(self, method, arg): member in TestCuda
985 self._spawn_method(test_method, 11)
|
| H A D | test_torch.py | 9486 def _spawn_method(self, method, arg): function
9503 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., -1., 1.]))
9504 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., inf, 1.]))
9505 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., -inf, 1.]))
9506 _spawn_method(_test_multinomial_invalid_probs, torch.tensor([1., 1., nan]))
|