Searched refs:test_pin_memory (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/test/ | ||
H A D | test_cuda_primary_ctx.py | 71 def test_pin_memory(self): member in TestCudaPrimaryCtx |
H A D | test_dataloader.py | 3081 def test_pin_memory(self): member in TestDictDataLoader |
H A D | test_nestedtensor.py | 5162 def test_pin_memory(self, device): member in TestNestedTensorSubclass |
H A D | test_autograd.py | 11502 def test_pin_memory(self, device): member in TestAutogradDeviceType |
H A D | test_torch.py | 8433 def test_pin_memory(self): member in TestTorch |