Make exported MetaGraphDefs deterministic

Nondeterministic serialized protos hurt caching. PiperOrigin-RevId: 301737437 Change-Id: Ia354d81bac2f2140d0ca84feca85d70a03c25a67
2020-03-18 22:01:39 -07:00 · 2020-03-18 22:01:39 -07:00 · f3e1c5c94d
commit f3e1c5c94d
parent 7fb2f93d40
1 changed files with 2 additions and 1 deletions
--- a/tensorflow/python/framework/graph_io.py
+++ b/tensorflow/python/framework/graph_io.py
@ -71,5 +71,6 @@ def write_graph(graph_or_graph_def, logdir, name, as_text=True):
                                        text_format.MessageToString(
                                            graph_def, float_format=''))
  else:
-    file_io.atomic_write_string_to_file(path, graph_def.SerializeToString())
+    file_io.atomic_write_string_to_file(
+        path, graph_def.SerializeToString(deterministic=True))
  return path