Searched refs:test_tensor_device (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_cuda_multigpu.py | 705 def test_tensor_device(self): member in TestCudaMultiGPU |
| H A D | test_tensor_creation_ops.py | 1856 def test_tensor_device(self, devices): member in TestTensorCreation |
| H A D | test_jit.py | 4677 def test_tensor_device(self): member in TestScript |