diff --git a/python/tvm/relax/frontend/onnx/onnx_frontend.py b/python/tvm/relax/frontend/onnx/onnx_frontend.py index 3b94ba1d6672..2e4e7a3125e9 100644 --- a/python/tvm/relax/frontend/onnx/onnx_frontend.py +++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py @@ -340,6 +340,8 @@ def base_impl(cls, bb, inputs, attr, params): x = _to_numpy(inputs[0]) y = _to_numpy(inputs[1]) output = cls.numpy_op(x, y) # pylint: disable=not-callable + if isinstance(x, relax.PrimValue) and isinstance(y, relax.PrimValue): + return relax.PrimValue(output.item()) if x.dtype == y.dtype: # no numpy precision widening output = output.astype(x.dtype)