Skip to content

Commit

Permalink
One more input shape fix.
Browse files Browse the repository at this point in the history
  • Loading branch information
gustavla committed Sep 21, 2023
1 parent 990711c commit 641fe7c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/onnx/test_pytorch_onnx_onnxruntime.py
Original file line number Diff line number Diff line change
Expand Up @@ -13148,7 +13148,7 @@ def forward(self, x):
model = torch.ao.quantization.convert(model)

# Set fixed input to avoid flaky test.
input = _construct_tensor_for_quantization_test((3, 2), offset=-384, max_val=12)
input = _construct_tensor_for_quantization_test((3, 4), offset=-384, max_val=12)
self.run_test(model, input)

@skipIfUnsupportedMinOpsetVersion(10)
Expand Down

0 comments on commit 641fe7c

Please sign in to comment.