Skip to content

Commit cf575b8

Browse files
authored
[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
1 parent 9fd3461 commit cf575b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tvm/relay/frontend/onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6021,7 +6021,7 @@ class Multinomial(OnnxOpConverter):
60216021

60226022
@classmethod
60236023
def _impl_v7(cls, inputs, attr, params):
6024-
dtype = attr.get("dtype", "int64")
6024+
dtype = attr.get("dtype", "int32")
60256025
sample_size = attr.get("sample_size", 1)
60266026
seed = attr.get("seed", None)
60276027
if seed is None:

0 commit comments

Comments
 (0)