Skip to content

Commit 2da5ccf

Browse files
committed
fix onnx test
1 parent 35c2f45 commit 2da5ccf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python/frontend/onnx/test_forward.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2993,7 +2993,7 @@ def forward(self, input):
29932993
torch.onnx.export(layer, dummy_input, file_name, export_params=True)
29942994

29952995
onnx_model = onnx.load(file_name)
2996-
relay.frontend.from_onnx(onnx_model, {"0": input_size})
2996+
relay.frontend.from_onnx(onnx_model, {"onnx::Reshape_0": input_size})
29972997

29982998

29992999
@tvm.testing.parametrize_targets

0 commit comments

Comments
 (0)