fixins

apache · Oct 14, 2022 · 1e29caa · 1e29caa
1 parent 45b894a
commit 1e29caa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: