diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-11-17 09:48:44 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-17 10:05:34 -0800 |
commit | 5da8731b128aed7f8434750984d30ae6b1b0e95b (patch) | |
tree | bac6141c6298d3e0038f1a9dfd3cff5baeb4e3d4 /tensorflow/python/framework/meta_graph.py | |
parent | 9a48456cc35f1887faa7cdc910b3fac98bad69d8 (diff) |
Adding __version__ and __git_version__ information into saved meta graphs.
Change: 139471934
Diffstat (limited to 'tensorflow/python/framework/meta_graph.py')
-rw-r--r-- | tensorflow/python/framework/meta_graph.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tensorflow/python/framework/meta_graph.py b/tensorflow/python/framework/meta_graph.py index 6f79e9ab20..6fd93f44bc 100644 --- a/tensorflow/python/framework/meta_graph.py +++ b/tensorflow/python/framework/meta_graph.py @@ -34,6 +34,7 @@ from tensorflow.core.protobuf import saver_pb2 from tensorflow.python.framework import importer from tensorflow.python.framework import op_def_registry from tensorflow.python.framework import ops +from tensorflow.python.framework import versions from tensorflow.python.lib.io import file_io from tensorflow.python.platform import tf_logging as logging from tensorflow.python.training import training_util @@ -351,8 +352,13 @@ def create_meta_graph_def(meta_info_def=None, # Creates a MetaGraphDef proto. meta_graph_def = meta_graph_pb2.MetaGraphDef() # Adds meta_info_def. - if meta_info_def: - meta_graph_def.meta_info_def.MergeFrom(meta_info_def) + if not meta_info_def: + meta_info_def = meta_graph_pb2.MetaGraphDef.MetaInfoDef() + + # Set the tf version strings to the current tf build. + meta_info_def.tensorflow_version = versions.__version__ + meta_info_def.tensorflow_git_version = versions.__git_version__ + meta_graph_def.meta_info_def.MergeFrom(meta_info_def) # Adds graph_def or the default. if not graph_def: |