| 243 | # but support for compiled modes is under development. |
| 244 | |
| 245 | class Foo(torch.nn.Module): |
| 246 | def __init__(self): |
| 247 | super().__init__() |
| 248 | self.l1 = torch.nn.Linear(4, 4) |
| 249 | self.l2 = torch.nn.Linear(4, 4) |
| 250 | |
| 251 | def forward(self, x): |
| 252 | return self.l2(self.l1(x)) |
| 253 | |
| 254 | class Bar(torch.nn.Module): |
| 255 | def __init__(self): |