Skip to content

Commit d8bc10b

Browse files
Added f to the list of TF function attributes
1 parent 19d3f97 commit d8bc10b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/tf_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ def tflist_to_onnx(g, shape_override, const_node_values=None):
203203
elif a == "output_shapes":
204204
# we should not need it since we pull the shapes above already
205205
pass
206-
elif a in {"body", "cond", "then_branch", "else_branch"}:
206+
elif a in {"body", "cond", "then_branch", "else_branch", "f"}:
207207
input_shapes = [inp.get_shape() for inp in node.inputs]
208208
nattr = get_tf_node_attr(node, a)
209209
attr[a] = nattr.name

0 commit comments

Comments
 (0)