Merge pull request #1094 from onnx/tom/FixSatefulPartitionedCallError

Added f to the list of TF function attributes

Author: guschmue

tf2onnx/tf_utils.py

@@ -203,7 +203,7 @@ def tflist_to_onnx(g, shape_override, const_node_values=None):
             elif a == "output_shapes":
                 # we should not need it since we pull the shapes above already
                 pass
-            elif a in {"body", "cond", "then_branch", "else_branch"}:
+            elif a in {"body", "cond", "then_branch", "else_branch", "f"}:
                 input_shapes = [inp.get_shape() for inp in node.inputs]
                 nattr = get_tf_node_attr(node, a)
                 attr[a] = nattr.name