Compare commits

...

1 Commits

Author SHA1 Message Date
099216fbc1 Test nn.linear bias
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
2024-05-15 15:01:59 -07:00

View File

@ -1764,9 +1764,10 @@ class TestQuantizeFx(QuantizationTestCase):
def __init__(self):
super().__init__()
self.linear = torch.nn.Linear(5, 10)
self.bias = torch.tensor((5,))
def forward(self, x):
return self.linear(x)
return self.linear(xi, bias=self.bias)
linear_module_input = torch.rand(8, 5)