Searched refs:CompModule (Results 1 – 1 of 1) sorted by relevance
645 class CompModule(torch.nn.Module): class in CompModuleTestWithCompiler680 self.module = CompModuleTestWithCompiler.CompModule(lowered_add, lowered_sub)682 CompModuleTestWithCompiler.CompModule(lowered_add, lowered_sub)744 class CompModule(torch.nn.Module): class in CompModuleTestSameNameWithCompiler776 self.module = CompModule() # noqa: F821