Searched refs:TestLinear (Results 1 – 7 of 7) sorted by relevance
44 class TestLinear(torch.nn.Module): class90 return TestLinear._get_random_inputs() + TestSDPA._get_random_inputs()632 self._test_model_with_non_decomp_partitioner(TestLinear())643 class TestLinear(torch.nn.Module): class657 model = TestLinear()746 model = TestLinear()
106 class TestLinear(unittest.TestCase): class
63 TEST(TestXNNPackOps, TestLinear) { in TEST() argument
16 class TestLinear(TestMPS): class
167 class TestLinear(unittest.TestCase): class
13075 class TestLinear(torch.nn.Module): class13099 self.fc1 = TestLinear(10, 10)13102 self.fc2 = TestLinear(10, 10)
3798 TEST_F(LazyOpsTest, TestLinear) { in TEST_F() argument