Searched defs:test_cuda (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ | ||
| H A D | utils.py | 337 def decorate_test_fn(test_fn, test_cuda, has_impl_parity, device): argument |
| /aosp_15_r20/external/pytorch/test/distributed/ | ||
| H A D | test_c10d_spawn_gloo.py | 173 def test_cuda(self): member in DistributedDataParallelSingleProcessTest |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ | ||
| H A D | common_nn.py | 3504 def test_cuda(self, test_case): member in ModuleTest |