add onnx checker

apache · May 8, 2019 · 59601d3 · 59601d3
1 parent e157728
commit 59601d3
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py
@@ -1133,6 +1133,13 @@ def from_onnx(model,
     params : dict of str to tvm.NDArray
         The parameter dict to be used by relay
     """
+    try:
+        from onnx import checker
+        if hasattr(checker, 'check_model'):
+            # try use onnx's own model checker before converting any model
+            checker.check_model(model)
+    except ImportError:
+            pass
     g = GraphProto(shape, dtype)
     graph = model.graph
     try: