Skip to content

Commit ec78417

Browse files
committed
replace fifoqueue with placeholder
1 parent a2dbc7b commit ec78417

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/onnx_opset/controlflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -372,14 +372,14 @@ def version_9(cls, ctx, node, **kwargs):
372372
ctx.copy_dtype(node.output[0], transpose_node.output[0])
373373

374374

375-
@tf_op("IteratorV2")
375+
@tf_op("IteratorV2", "FIFOQueueV2")
376376
class Iterator:
377377
@classmethod
378378
def version_8(cls, ctx, node, **kwargs):
379379
ctx.remove_node(node.name)
380380

381381

382-
@tf_op("IteratorGetNext")
382+
@tf_op("IteratorGetNext", "QueueDequeueV2")
383383
class IteratorGetNext:
384384
@classmethod
385385
def version_8(cls, ctx, node, **kwargs):

0 commit comments

Comments
 (0)