Skip to content

Commit ff634a7

Browse files
committed
fix bug found when option "custom-ops" is used
1 parent 9d22804 commit ff634a7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/convert.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def main():
7575

7676
if args.custom_ops:
7777
# default custom ops for tensorflow-onnx are in the "tf" namespace
78-
custom_ops = {op: default_custom_op_handler for op in args.custom_ops.split(",")}
78+
custom_ops = {op: (default_custom_op_handler, []) for op in args.custom_ops.split(",")}
7979
extra_opset = [helper.make_opsetid(_TENSORFLOW_DOMAIN, 1)]
8080
else:
8181
custom_ops = {}

0 commit comments

Comments
 (0)