Searched defs:test_memory_format (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_modules.py | 619 def test_memory_format(self, device, dtype, module_info, training): member in TestModule |
| H A D | test_testing.py | 1446 def test_memory_format(self, dtype, device, memory_format_and_shape): member in TestMakeTensor |
| H A D | test_torch.py | 8635 def test_memory_format(self): member in TestTorch |