We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
mapping.TENSOR_TYPE_TO_NP_TYPE
helper.tensor_dtype_to_np_dtype
1 parent fbd9fe4 commit 88d10edCopy full SHA for 88d10ed
python/tvm/relax/frontend/onnx/onnx_frontend.py
@@ -60,13 +60,13 @@ def get_type(elem_type: Union[str, int]) -> str:
60
return elem_type
61
62
try:
63
- from onnx.mapping import ( # pylint: disable=import-outside-toplevel
64
- TENSOR_TYPE_TO_NP_TYPE,
+ from onnx.helper import ( # pylint: disable=import-outside-toplevel
+ tensor_dtype_to_np_dtype,
65
)
66
except ImportError as exception:
67
raise ImportError("Unable to import onnx which is required {}".format(exception))
68
69
- return str(TENSOR_TYPE_TO_NP_TYPE[elem_type])
+ return str(tensor_dtype_to_np_dtype[elem_type])
70
71
72
def get_constant(
0 commit comments