From 3ffc3685377ef4f346b7c42656642bdc8e0f73fc Mon Sep 17 00:00:00 2001 From: Andrew Zhao Luo Date: Thu, 13 Oct 2022 11:15:42 -0700 Subject: [PATCH 1/2] onnx get right import --- python/tvm/relay/frontend/onnx.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 84a5fc3b8237..9ada97842bb3 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -5906,7 +5906,16 @@ def from_onnx( graph = model.graph try: - opset_in_model = model.opset_import[0].version if model.opset_import else 1 + opset_in_model = 1 + if model.opset_import: + # TODO: for now we only really support ai.onnx op set + # TODO: handle other namespaces well see https://github.com/apache/tvm/issues/10950 + for opset_identifier in model.opset_import: + # As per https://github.com/onnx/onnx/blob/main/docs/IR.md + # All operator sets except the default one must specify the operator version + if str(opset_identifier.name) in ["ai.onnx", ""]: + opset_in_model = opset_identifier.version + break except AttributeError: opset_in_model = 1 From e4a0c87ab616c3da5a8205b0d109603e45b5f694 Mon Sep 17 00:00:00 2001 From: Andrew Luo Date: Fri, 14 Oct 2022 11:13:39 -0700 Subject: [PATCH 2/2] fixins --- python/tvm/relay/frontend/onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 9ada97842bb3..8c4c056221f6 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -5913,7 +5913,7 @@ def from_onnx( for opset_identifier in model.opset_import: # As per https://github.com/onnx/onnx/blob/main/docs/IR.md # All operator sets except the default one must specify the operator version - if str(opset_identifier.name) in ["ai.onnx", ""]: + if str(opset_identifier.domain) in ["ai.onnx", ""]: opset_in_model = opset_identifier.version break except AttributeError: