(self, x)
| 852 | self.bn = torch.nn.BatchNorm2d(3) |
| 853 | |
| 854 | def forward(self, x): |
| 855 | x = self.conv(x) |
| 856 | x = self.bn(x) |
| 857 | return (x,) |
| 858 | |
| 859 | ep_for_training = torch.export.export(DecompExample(), (torch.randn(1, 1, 3, 3),)) |
| 860 | print(ep_for_training.graph) |
nothing calls this directly
no outgoing calls
no test coverage detected