Skip to content

Commit afab9d7

Browse files
authored
Merge pull request #587 from lucienwang1009/onehot_bug
typo
2 parents 13fb5a4 + ef9bac1 commit afab9d7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -974,7 +974,7 @@ def version_9(cls, ctx, node, **kwargs):
974974
if ctx.is_target(constants.TARGET_RS6) \
975975
and output_dtype not in [onnx_pb.TensorProto.INT64, onnx_pb.TensorProto.INT32]:
976976
logger.warning("unsupported dtype in onnxruntime, onehot-9 can't be used directly")
977-
cls.version_5(ctx, node, **kwargs)
977+
cls.version_1(ctx, node, **kwargs)
978978
return
979979

980980
depth = node.input[1]

0 commit comments

Comments
 (0)