Merge pull request onnx#1047 from onnx/tom/QueueDequeueUpToV2

Implemented QueueDequeueUpToV2
buptqq · Aug 7, 2020 · 4a6125b · 4a6125b
2 parents 55e6d96 + 2e91bab
commit 4a6125b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tf2onnx/onnx_opset/generator.py b/tf2onnx/onnx_opset/generator.py
@@ -197,7 +197,7 @@ def version_8(cls, ctx, node, **kwargs):
         ctx.add_graph_input(output_names[1], type_1, shape_1)
 
 
-@tf_op("QueueDequeueManyV2")
+@tf_op("QueueDequeueManyV2", "QueueDequeueUpToV2")
 class QueueDequeueManyV2:
     @classmethod
     def version_8(cls, ctx, node, **kwargs):