Searched refs:TestModelWithExport (Results 1 – 1 of 1) sorted by relevance
143 class TestModelWithExport(torch.nn.Module): class154 make_global(TestModelWithExport)155 pdt_model = TestModelWithExport()