diff options
author | 2018-09-13 13:59:24 -0700 | |
---|---|---|
committer | 2018-09-13 14:10:37 -0700 | |
commit | 304faf0444260912b6996d39227417c09561c37e (patch) | |
tree | d7cbb93df79cf2def0c3a76541b5afb087c6a09b /tensorflow/python/tools | |
parent | 490e46f29dba0254fa69385d4235ab26854868c8 (diff) |
Remove tf.contrib.get_signature_def_by_key. This can be replaced by meta_graph_def.signature_def[signature_def_key]
PiperOrigin-RevId: 212873314
Diffstat (limited to 'tensorflow/python/tools')
-rw-r--r-- | tensorflow/python/tools/saved_model_cli.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/tensorflow/python/tools/saved_model_cli.py b/tensorflow/python/tools/saved_model_cli.py index c5289564fe..d8ba13d8d2 100644 --- a/tensorflow/python/tools/saved_model_cli.py +++ b/tensorflow/python/tools/saved_model_cli.py @@ -33,7 +33,6 @@ import numpy as np from six import integer_types from tensorflow.contrib.saved_model.python.saved_model import reader -from tensorflow.contrib.saved_model.python.saved_model import signature_def_utils from tensorflow.core.example import example_pb2 from tensorflow.core.framework import types_pb2 from tensorflow.python.client import session @@ -97,8 +96,7 @@ def _get_inputs_tensor_info_from_meta_graph_def(meta_graph_def, Returns: A dictionary that maps input tensor keys to TensorInfos. """ - return signature_def_utils.get_signature_def_by_key(meta_graph_def, - signature_def_key).inputs + return meta_graph_def.signature_def[signature_def_key].inputs def _get_outputs_tensor_info_from_meta_graph_def(meta_graph_def, @@ -116,8 +114,7 @@ def _get_outputs_tensor_info_from_meta_graph_def(meta_graph_def, Returns: A dictionary that maps output tensor keys to TensorInfos. """ - return signature_def_utils.get_signature_def_by_key(meta_graph_def, - signature_def_key).outputs + return meta_graph_def.signature_def[signature_def_key].outputs def _show_inputs_outputs(saved_model_dir, tag_set, signature_def_key, indent=0): |