This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new c725954 fix onnx test
c725954 is described below
commit c725954fd040bf797538856e0c448c875fd9c07f
Author: Masahiro Masuda <[email protected]>
AuthorDate: Thu Mar 31 19:34:29 2022 +0900
fix onnx test
---
tests/python/frontend/onnx/test_forward.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/python/frontend/onnx/test_forward.py
b/tests/python/frontend/onnx/test_forward.py
index 91775d2..8805d4d 100644
--- a/tests/python/frontend/onnx/test_forward.py
+++ b/tests/python/frontend/onnx/test_forward.py
@@ -2993,7 +2993,7 @@ def test_unsqueeze_constant(target, dev):
torch.onnx.export(layer, dummy_input, file_name, export_params=True)
onnx_model = onnx.load(file_name)
- relay.frontend.from_onnx(onnx_model, {"0": input_size})
+ relay.frontend.from_onnx(onnx_model, {"onnx::Reshape_0": input_size})
@tvm.testing.parametrize_targets
