From 33d4f68722e7b61c1d475a993472669254b87825 Mon Sep 17 00:00:00 2001 From: sleepwalker2017 <1028552010@qq.com> Date: Thu, 30 Jul 2020 01:30:03 +0800 Subject: [PATCH] fix bug when converting constant nodes with types of int64 or float64 (#6159) Co-authored-by: yuweilong --- python/tvm/relay/frontend/tensorflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/tensorflow.py b/python/tvm/relay/frontend/tensorflow.py index aa62702b22146..a06b0ca425cf0 100644 --- 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,