From b34eed57107f2f505c82092da128f96a185c7ad9 Mon Sep 17 00:00:00 2001 From: Danni <38550500+isdanni@users.noreply.github.com> Date: Sun, 2 Apr 2023 17:38:31 -0700 Subject: [PATCH] Update oneflow.py --- python/tvm/relay/frontend/oneflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relay/frontend/oneflow.py b/python/tvm/relay/frontend/oneflow.py index 1aba9e64190f..07bb76871ab1 100644 --- a/python/tvm/relay/frontend/oneflow.py +++ b/python/tvm/relay/frontend/oneflow.py @@ -84,8 +84,8 @@ def get_node_info(node): shape = tuple(node.input_conf.blob_conf.shape.dim) # get data type dtype = node.input_conf.blob_conf.data_type - if dtype in list(FLOW_2_NP_DTYPE.keys()): - data_type = FLOW_2_NP_DTYPE[dtype] + if dtype in list(FLOW_2_STR_DTYPE.keys()): + data_type = FLOW_2_STR_DTYPE[dtype] else: raise IndexError("Please check the data type of your node: %s" % node.name)