Skip to content

Commit 0110037

Browse files
Merge pull request #1050 from onnx/tom/FixQueueDequeueUpToV2Error
Added brackets to fix error in QueueDequeueUpToV2
2 parents c867e52 + 3ead413 commit 0110037

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def version_8(cls, ctx, node, **kwargs):
197197
ctx.add_graph_input(output_names[1], type_1, shape_1)
198198

199199

200-
@tf_op("QueueDequeueManyV2", "QueueDequeueUpToV2")
200+
@tf_op(["QueueDequeueManyV2", "QueueDequeueUpToV2"])
201201
class QueueDequeueManyV2:
202202
@classmethod
203203
def version_8(cls, ctx, node, **kwargs):

0 commit comments

Comments
 (0)