Lines Matching full:memory_format

242         for shapes, memory_format in pairs:
245 …             expected = torch.empty(shape, device=device, dtype=dtype, memory_format=memory_format)
246                 actual = refs.empty(shape, device=device, dtype=dtype, memory_format=memory_format)
251                 expected = torch.clone(a, memory_format=memory_format)
252                 actual = torch.clone(a, memory_format=memory_format)
257                 expected = a.contiguous(memory_format=memory_format)
258                 actual = refs.contiguous(a, memory_format=memory_format)
363             a = torch.zeros([2] * ndim).to(memory_format=mf)
365             self.assertTrue(res.is_contiguous(memory_format=mf))
371         self.assertTrue(a.is_contiguous(memory_format=torch.channels_last))
376         self.assertTrue(actual.is_contiguous(memory_format=torch.channels_last))
381         self.assertTrue(a.is_contiguous(memory_format=torch.channels_last))