| 53 | |
| 54 | |
| 55 | class SimpleConv(torch.nn.Module): |
| 56 | def __init__(self) -> None: |
| 57 | super().__init__() |
| 58 | self.conv = torch.nn.Conv2d( |
| 59 | in_channels=3, out_channels=16, kernel_size=3, padding=1 |
| 60 | ) |
| 61 | self.relu = torch.nn.ReLU() |
| 62 | |
| 63 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 64 | a = self.conv(x) |
| 65 | return self.relu(a) |
| 66 | |
| 67 | |
| 68 | example_args: tuple[torch.Tensor] = (torch.randn(1, 3, 256, 256),) |
no outgoing calls
no test coverage detected