Merge pull request #427 from pengwa/add_update_proto

add update_proto() before  return Graph to users

Author: pengwa

tf2onnx/graph.py

@@ -105,6 +105,7 @@ def name(self):
 
     @property
     def op(self):
+        """TODO: have a better interface for this."""
         return self._op
 
     @property

tf2onnx/optimizer/__init__.py

@@ -31,6 +31,8 @@ def optimize_graph(graph, debug=False):
         opts = _get_optimizers()
         for opt in opts.values():
             graph = opt(debug=debug).optimize(graph)
+
+        graph.update_proto()
         return graph
     except Exception:
         # degradation to non-optimized model proto

tf2onnx/tfonnx.py

@@ -2568,6 +2568,8 @@ def process_tf_graph(tf_graph, continue_on_error=False, verbose=False, target=No
     # onnx requires topological sorting
     topological_sort(g, continue_on_error)
 
+    g.update_proto()
+
     if verbose:
         print("tensorflow ops: {}".format(op_cnt))
         print("tensorflow attr: {}".format(attr_cnt))