From 641fe7cb53bae2c7d9403ff739abd2f9c393fba0 Mon Sep 17 00:00:00 2001 From: Gustav Larsson Date: Wed, 20 Sep 2023 19:33:44 -0700 Subject: [PATCH] One more input shape fix. --- test/onnx/test_pytorch_onnx_onnxruntime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py index 7aef2feed22b9..72968f3334262 100644 --- a/test/onnx/test_pytorch_onnx_onnxruntime.py +++ b/test/onnx/test_pytorch_onnx_onnxruntime.py @@ -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)