| 284 | def test_perform_optimization_false(): |
| 285 | def _create_dummy_model(): |
| 286 | class MockModel(torch.nn.Module): |
| 287 | def __init__(self): |
| 288 | super(MockModel, self).__init__() |
| 289 | self.linear = torch.nn.Linear(10, 5) |
| 290 | |
| 291 | def forward(self, x): |
| 292 | return self.linear(x) |
| 293 | |
| 294 | model = MockModel() |
| 295 | dummy_input = torch.randn(1, 10) |
no outgoing calls