Searched refs:test_memory_format (Results 1 – 5 of 5) sorted by relevance
533 test_memory_format = [torch.contiguous_format, cl_format]537 test_memory_format,
3100 test_memory_format = [torch.contiguous_format]3106 test_memory_format.append(channels_last)3107 for memory_format in test_memory_format:
619 def test_memory_format(self, device, dtype, module_info, training): member in TestModule
1446 def test_memory_format(self, dtype, device, memory_format_and_shape): member in TestMakeTensor
8635 def test_memory_format(self): member in TestTorch