Skip to content

Commit d315796

Browse files
committed
ignore fifoqueue attr
1 parent ec78417 commit d315796

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tf2onnx/tf_utils.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def tflist_to_onnx(g, shape_override):
138138
"Tout", "Tlabels", "Tindex", "element_shape", "Targmax", "Tperm", "Tcond",
139139
"T_threshold", "element_dtype", "shape_type", "_lower_using_switch_merge",
140140
"parallel_iterations", "_num_original_outputs", "output_types", "output_shapes",
141-
"key_dtype", "value_dtype", "Tin", "Tout"]
141+
"key_dtype", "value_dtype", "Tin", "Tout", "capacity", "component_types", "shapes"]
142142

143143
node_list = g.get_operations()
144144
functions = {}
@@ -155,6 +155,11 @@ def tflist_to_onnx(g, shape_override):
155155

156156
# create dict with output to shape mappings
157157
for node in ops:
158+
'''
159+
if node.type.startswith('FIFOQueueV2') or node.type.startswith('QueueDequeueV2'):
160+
print ('attr:\n', node.node_def.attr, '\ninputs:\n', node.inputs, '\noutputs:\n', node.outputs)
161+
ii = input('ii')
162+
'''
158163
for out in node.outputs:
159164
shape = shape_override.get(out.name)
160165
if shape is None:

0 commit comments

Comments
 (0)