Class TestLinear

exir/program/test/test_program.py:46–57

class TestLinear(torch.nn.Module):
    def __init__(self):
        super().__init__()
        self.linear = torch.nn.Linear(32, 16, bias=True)

    def forward(self, x):
        return self.linear(x)

    @classmethod
    def _get_random_inputs(cls):
        x = torch.rand(8, 32)
        return (x,)


class TestSDPA(torch.nn.Module):