| 40 | |
| 41 | |
| 42 | class SerializationInterop(FileSetup): |
| 43 | path = 'ivalue.pt' |
| 44 | |
| 45 | def setup(self): |
| 46 | ones = torch.ones(2, 2) |
| 47 | twos = torch.ones(3, 5) * 2 |
| 48 | |
| 49 | value = (ones, twos) |
| 50 | |
| 51 | torch.save(value, self.path, _use_new_zipfile_serialization=True) |
| 52 | |
| 53 | |
| 54 | # See testTorchSaveError in test/cpp/jit/tests.h for usage |
no outgoing calls
no test coverage detected
searching dependent graphs…