fix bug when converting constant nodes with types of int64 or float64 (…

…apache#6159) Co-authored-by: yuweilong <yuweilong03@meituan.com>
trevor-m · Aug 26, 2020 · 33d4f68 · 33d4f68
1 parent 4533b78
commit 33d4f68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py
@@ -2835,7 +2835,7 @@ def _parse_param(self, key, value, name, shape):
 
             array_ndim = len(np_array.shape)
             if array_ndim == 0:
-                self._nodes[name] = [tvm.relay.const(np_array)]
+                self._nodes[name] = [tvm.relay.const(np_array, np_array.dtype)]
             else:
                 self._params[name] = tvm.nd.array(np_array)
                 self._nodes[name] = [_expr.var(name,