Merge pull request #968 from onnx/rashuai/QueueDequeueManyV2

support QueueDequeueManyV2

Author: RandySheriffH

tf2onnx/onnx_opset/generator.py

@@ -194,3 +194,15 @@ def version_8(cls, ctx, node, **kwargs):
         ctx.remove_node(node.name)
         ctx.add_graph_input(output_names[0], type_0, shape_0)
         ctx.add_graph_input(output_names[1], type_1, shape_1)
+
+
+@tf_op("QueueDequeueManyV2")
+class QueueDequeueManyV2:
+    @classmethod
+    def version_8(cls, ctx, node, **kwargs):
+        outputs = node.output
+        shapes = node.output_shapes
+        dtypes = node.output_dtypes
+        ctx.remove_node(node.name)
+        for i, output in enumerate(outputs):
+            ctx.add_graph_input(output, dtypes[i], shapes[i])