Searched refs:ModuleWithJit (Results 1 – 1 of 1) sorted by relevance
825 class ModuleWithJit(nn.Module): class841 self.module_with_jit_1 = ModuleWithJit()842 self.module_with_jit_2 = ModuleWithJit()