[v1.x] fix onnx type inference issue (#20130)

* fix type inference issue * Update _export_onnx.py
apache · Apr 8, 2021 · a2af352 · a2af352
1 parent 5b7826e
commit a2af352
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/mxnet/onnx/mx2onnx/_export_onnx.py b/python/mxnet/onnx/mx2onnx/_export_onnx.py
@@ -50,6 +50,7 @@
 import logging
 import json
 
+import numpy as np
 from mxnet import ndarray as nd
 
 
@@ -290,7 +291,7 @@ def create_onnx_graph_proto(self, sym, params, in_shapes, in_types, verbose=Fals
         class NodeOutput:
             def __init__(self, name, dtype):
                 self.name = name
-                self.dtype = dtype
+                self.dtype = np.dtype(dtype)
 
         initializer = []
         all_processed_nodes = []