This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new cf575b8f83 [Relay][ONNX] fix the wrong default value about dtype in
Multinomial converter (#16624)
cf575b8f83 is described below
commit cf575b8f837761d98e3b309eeedce1e0d4138bb8
Author: Qingchao Shen <[email protected]>
AuthorDate: Thu Feb 22 23:25:31 2024 +0800
[Relay][ONNX] fix the wrong default value about dtype in Multinomial
converter (#16624)
Update onnx.py
the default type of attribute 'dtype' is int32 rather than int64
---
python/tvm/relay/frontend/onnx.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/tvm/relay/frontend/onnx.py
b/python/tvm/relay/frontend/onnx.py
index 9a42fe2490..ddd0d34c5c 100644
--- a/python/tvm/relay/frontend/onnx.py
+++ b/python/tvm/relay/frontend/onnx.py
@@ -6021,7 +6021,7 @@ class Multinomial(OnnxOpConverter):
@classmethod
def _impl_v7(cls, inputs, attr, params):
- dtype = attr.get("dtype", "int64")
+ dtype = attr.get("dtype", "int32")
sample_size = attr.get("sample_size", 1)
seed = attr.get("seed", None)
if seed is None:
