(self)
| 208 | |
| 209 | class OptimizeNoForwardTest(torch.nn.Module): |
| 210 | def __init__(self): |
| 211 | super().__init__() |
| 212 | self.l = nn.Linear(10, 100) |
| 213 | self.l2 = nn.Linear(100, 1) |
| 214 | self.d = nn.Dropout(p=0.2) |
| 215 | |
| 216 | @torch.jit.export |
| 217 | def foo(self, x): |