| 130 | example_args = (torch.randn(1, 3, 10, 10, dtype=torch.float32),) |
| 131 | |
| 132 | class UnaryOp(Module): |
| 133 | def forward(self, input): |
| 134 | return pytorch_op(input) |
| 135 | |
| 136 | @tvm.script.ir_module |
| 137 | class expected: |
no outgoing calls
searching dependent graphs…