Lines Matching full:conv2d

20 class Conv2d(torch.nn.Module):  class
44         self.conv = torch.nn.Conv2d(
70         self.first = torch.nn.Conv2d(
77         self.second = torch.nn.Conv2d(
96         self.conv1 = torch.nn.Conv2d(
106         self.conv2 = torch.nn.Conv2d(
131         self.conv = torch.nn.Conv2d(
168             .check_count({"torch.ops.aten.conv2d": conv_count})
184             self._test(Conv2d(bias=has_bias, dtype=torch.float16))
188             self._test(Conv2d(bias=has_bias))
197                 Conv2d(bias=has_bias), quant_config=get_symmetric_quantization_config()
202             Conv2d(),
216             Conv2d(
228         self._test(Conv2d(groups=2, in_channels=2, out_channels=6))
232             Conv2d(groups=2, in_channels=2, out_channels=6),
240                 self.conv2d = torch.nn.Conv2d(in_features, out_features, kernel_size)
246                 y = self.conv2d(x)
272                 self.conv = torch.nn.Conv2d(
308                 self.conv1 = torch.nn.Conv2d(
348                 self.conv1 = torch.nn.Conv2d(
382                     torch.nn.Conv2d(1, 1, 1),
384                     torch.nn.Conv2d(1, 64, 1),
404                 self.conv1 = torch.nn.Conv2d(1, 1, 1)
405                 self.conv2 = torch.nn.Conv2d(1, 64, 1)