Searched refs:TwoLinearModule (Results 1 – 3 of 3) sorted by relevance
104 TestHelperModules.TwoLinearModule().eval(),182 TestHelperModules.TwoLinearModule().eval(),
129 m_eager = TestHelperModules.TwoLinearModule().eval()245 m_eager = TestHelperModules.TwoLinearModule().eval()543 m_eager = TestHelperModules.TwoLinearModule().eval()589 m_eager = TestHelperModules.TwoLinearModule().eval()636 m_eager = TestHelperModules.TwoLinearModule().eval()
2742 class TwoLinearModule(torch.nn.Module): class in TestHelperModules