Merge pull request #968 from onnx/rashuai/QueueDequeueManyV2

support QueueDequeueManyV2
onnx · Jun 9, 2020 · 9c8e9e1 · 9c8e9e1
2 parents 4bba41c + 0ca1d60
commit 9c8e9e1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tf2onnx/onnx_opset/generator.py b/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])