From d3dc90227c8eb985eb84243fa25f40d2e88781d1 Mon Sep 17 00:00:00 2001 From: Masahiro Hiramori Date: Mon, 26 May 2025 21:47:09 +0900 Subject: [PATCH] replace `mapping.TENSOR_TYPE_TO_NP_TYPE` with `helper.tensor_dtype_to_np_dtype` --- python/tvm/relax/frontend/onnx/onnx_frontend.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tvm/relax/frontend/onnx/onnx_frontend.py b/python/tvm/relax/frontend/onnx/onnx_frontend.py index f6bde18b8db5..bae308f435ce 100644 --- a/python/tvm/relax/frontend/onnx/onnx_frontend.py +++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py @@ -60,13 +60,13 @@ def get_type(elem_type: Union[str, int]) -> str: return elem_type try: - from onnx.mapping import ( # pylint: disable=import-outside-toplevel - TENSOR_TYPE_TO_NP_TYPE, + from onnx.helper import ( # pylint: disable=import-outside-toplevel + tensor_dtype_to_np_dtype, ) except ImportError as exception: raise ImportError("Unable to import onnx which is required {}".format(exception)) - return str(TENSOR_TYPE_TO_NP_TYPE[elem_type]) + return str(tensor_dtype_to_np_dtype(elem_type)) def get_constant(