Lines Matching full:memory_format

73         for memory_format, enabled in [
92                           groups=groups).to(memory_format=memory_format)
93 …            x = torch.randn(batch_size, iC, input_size, input_size).to(memory_format=memory_format)
114         for memory_format, enabled in [
121 …                   m = M(unary_fn, 3, oC, bias, kernel_size=(3, 3)).to(memory_format=memory_format)
122                         x = torch.randn(1, 3, 224, 224).to(memory_format=memory_format)
141         for module, dim, memory_format in [
164                   groups=groups).to(memory_format=memory_format)
168             x = torch.randn(input_sizes).to(memory_format=memory_format)
245                 for bias, dilation, groups, memory_format in options:
249                     x = torch.randn(x_shape, dtype=torch.float32).to(memory_format=memory_format)
251                     mod = mod.to(memory_format=memory_format).eval()
281                 for fuse_relu, bias, dilation, groups, memory_format in options:
285                     x = torch.randn(x_shape, dtype=torch.float32).to(memory_format=memory_format)
287                     mod = mod.to(memory_format=memory_format).eval()
363                 for bias, dilation, groups, memory_format, prepack_weight in options:
367                     x = torch.randn(x_shape, dtype=torch.float32).to(memory_format=memory_format)
369                     mod = mod.to(memory_format=memory_format).eval()