| 1800 | |
| 1801 | def test_batchnorm2d(): |
| 1802 | class BatchNorm2d1(Module): |
| 1803 | def __init__(self): |
| 1804 | super().__init__() |
| 1805 | self.bn = torch.nn.BatchNorm2d(3) |
| 1806 | |
| 1807 | def forward(self, input): |
| 1808 | return self.bn(input) |
| 1809 | |
| 1810 | @tvm.script.ir_module |
| 1811 | class expected1: |
no outgoing calls
searching dependent graphs…