Lines Matching full:memory_format

95     def _test_dropout(self, cls, device, input, memory_format=torch.contiguous_format):  argument
100         input_var = input.clone(memory_format=memory_format).requires_grad_()
102         self.assertTrue(output.is_contiguous(memory_format=memory_format))
105         self.assertTrue(input_var.grad.is_contiguous(memory_format=memory_format))
109         input_var = input.clone(memory_format=memory_format).requires_grad_()
111         self.assertTrue(output.is_contiguous(memory_format=memory_format))
114         self.assertTrue(input_var.grad.is_contiguous(memory_format=memory_format))
127         self, cls, device, memory_format=torch.contiguous_format  argument
137                 2, 3, 3, 6, device=device, memory_format=memory_format
145                 self.assertTrue(out.is_contiguous(memory_format=memory_format))
178             nn.Dropout, device, memory_format=torch.channels_last
249             nn.Dropout2d, device, input, memory_format=torch.channels_last
254             nn.Dropout2d, device, memory_format=torch.channels_last
293             nn.Dropout3d, device, memory_format=torch.channels_last