diff options
author | Anna R <annarev@google.com> | 2018-10-01 11:53:27 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 11:58:12 -0700 |
commit | f1fd53748b99532b2572b8909efcd4f5c06ce28d (patch) | |
tree | 16ae93334d4b10a3bf7cfb1f088efd725857fb54 /tensorflow/python/saved_model | |
parent | 7cabc6be4e32dfb7f42c7f5e33549984bfdb68a3 (diff) |
Updating function and class tf_export decorators for endpoints according to
https://github.com/tensorflow/community/pull/16.
In addition to the changes in the doc, I made the following updates (these changes make sense to me and I didn't notice them when compiling the doc):
* deprecate saved_model.builder.SavedModelBuilder - replaced with saved_model.SavedModelBuilder
* deprecate python_io.tf_record_iterator - replaced with io.tf_record_iterator
* deprecate python_io.TFRecordWriter - replaced with io.TFRecordWriter
* move reduce_join to tf.string
PiperOrigin-RevId: 215253944
Diffstat (limited to 'tensorflow/python/saved_model')
-rw-r--r-- | tensorflow/python/saved_model/builder_impl.py | 7 | ||||
-rw-r--r-- | tensorflow/python/saved_model/loader_impl.py | 8 | ||||
-rw-r--r-- | tensorflow/python/saved_model/main_op_impl.py | 5 | ||||
-rw-r--r-- | tensorflow/python/saved_model/signature_def_utils_impl.py | 27 | ||||
-rw-r--r-- | tensorflow/python/saved_model/utils_impl.py | 10 |
5 files changed, 44 insertions, 13 deletions
diff --git a/tensorflow/python/saved_model/builder_impl.py b/tensorflow/python/saved_model/builder_impl.py index 8e7f123a85..8bf057f69d 100644 --- a/tensorflow/python/saved_model/builder_impl.py +++ b/tensorflow/python/saved_model/builder_impl.py @@ -36,10 +36,13 @@ from tensorflow.python.saved_model import utils_impl as saved_model_utils from tensorflow.python.training import saver as tf_saver from tensorflow.python.util import compat from tensorflow.python.util.deprecation import deprecated_args +from tensorflow.python.util.deprecation import deprecated_endpoints from tensorflow.python.util.tf_export import tf_export -@tf_export("saved_model.builder.SavedModelBuilder") +@tf_export("saved_model.Builder", + "saved_model.builder.SavedModelBuilder") +@deprecated_endpoints("saved_model.builder.SavedModelBuilder") class SavedModelBuilder(object): """Builds the `SavedModel` protocol buffer and saves variables and assets. @@ -61,7 +64,7 @@ class SavedModelBuilder(object): Typical usage for the `SavedModelBuilder`: ```python ... - builder = tf.saved_model.builder.SavedModelBuilder(export_dir) + builder = tf.saved_model.Builder(export_dir) with tf.Session(graph=tf.Graph()) as sess: ... diff --git a/tensorflow/python/saved_model/loader_impl.py b/tensorflow/python/saved_model/loader_impl.py index e8536108e8..895644a030 100644 --- a/tensorflow/python/saved_model/loader_impl.py +++ b/tensorflow/python/saved_model/loader_impl.py @@ -34,6 +34,7 @@ from tensorflow.python.saved_model import constants from tensorflow.python.saved_model import utils_impl as saved_model_utils from tensorflow.python.training import saver as tf_saver from tensorflow.python.util import compat +from tensorflow.python.util import deprecation from tensorflow.python.util.tf_export import tf_export @@ -144,7 +145,10 @@ def _get_main_op_tensor( return main_op_tensor -@tf_export("saved_model.loader.maybe_saved_model_directory") +@tf_export("saved_model.maybe_saved_model_directory", + "saved_model.loader.maybe_saved_model_directory") +@deprecation.deprecated_endpoints( + "saved_model.loader.maybe_saved_model_directory") def maybe_saved_model_directory(export_dir): """Checks whether the provided export directory could contain a SavedModel. @@ -165,7 +169,7 @@ def maybe_saved_model_directory(export_dir): return file_io.file_exists(txt_path) or file_io.file_exists(pb_path) -@tf_export("saved_model.loader.load") +@tf_export("saved_model.load", "saved_model.loader.load") def load(sess, tags, export_dir, import_scope=None, **saver_kwargs): """Loads the model from a SavedModel as specified by tags. diff --git a/tensorflow/python/saved_model/main_op_impl.py b/tensorflow/python/saved_model/main_op_impl.py index 631ee63729..ad4511b28e 100644 --- a/tensorflow/python/saved_model/main_op_impl.py +++ b/tensorflow/python/saved_model/main_op_impl.py @@ -22,6 +22,7 @@ from tensorflow.python.framework import ops from tensorflow.python.ops import control_flow_ops from tensorflow.python.ops import lookup_ops from tensorflow.python.ops import variables +from tensorflow.python.util import deprecation from tensorflow.python.util.tf_export import tf_export @@ -42,7 +43,9 @@ def main_op(): # TODO(sukritiramesh): Integrate with Saver for complete restore functionality. -@tf_export('saved_model.main_op.main_op_with_restore') +@tf_export('saved_model.main_op_with_restore', + 'saved_model.main_op.main_op_with_restore') +@deprecation.deprecated_endpoints('saved_model.main_op.main_op_with_restore') def main_op_with_restore(restore_op_name): """Returns a main op to init variables, tables and restore the graph. diff --git a/tensorflow/python/saved_model/signature_def_utils_impl.py b/tensorflow/python/saved_model/signature_def_utils_impl.py index 37f927f381..a1034416e9 100644 --- a/tensorflow/python/saved_model/signature_def_utils_impl.py +++ b/tensorflow/python/saved_model/signature_def_utils_impl.py @@ -24,10 +24,14 @@ from tensorflow.core.protobuf import meta_graph_pb2 from tensorflow.python.framework import ops from tensorflow.python.saved_model import signature_constants from tensorflow.python.saved_model import utils +from tensorflow.python.util import deprecation from tensorflow.python.util.tf_export import tf_export -@tf_export('saved_model.signature_def_utils.build_signature_def') +@tf_export('saved_model.build_signature_def', + 'saved_model.signature_def_utils.build_signature_def') +@deprecation.deprecated_endpoints( + 'saved_model.signature_def_utils.build_signature_def') def build_signature_def(inputs=None, outputs=None, method_name=None): """Utility function to build a SignatureDef protocol buffer. @@ -53,7 +57,10 @@ def build_signature_def(inputs=None, outputs=None, method_name=None): return signature_def -@tf_export('saved_model.signature_def_utils.regression_signature_def') +@tf_export('saved_model.regression_signature_def', + 'saved_model.signature_def_utils.regression_signature_def') +@deprecation.deprecated_endpoints( + 'saved_model.signature_def_utils.regression_signature_def') def regression_signature_def(examples, predictions): """Creates regression signature from given examples and predictions. @@ -95,7 +102,10 @@ def regression_signature_def(examples, predictions): return signature_def -@tf_export('saved_model.signature_def_utils.classification_signature_def') +@tf_export('saved_model.classification_signature_def', + 'saved_model.signature_def_utils.classification_signature_def') +@deprecation.deprecated_endpoints( + 'saved_model.signature_def_utils.classification_signature_def') def classification_signature_def(examples, classes, scores): """Creates classification signature from given examples and predictions. @@ -148,7 +158,10 @@ def classification_signature_def(examples, classes, scores): return signature_def -@tf_export('saved_model.signature_def_utils.predict_signature_def') +@tf_export('saved_model.predict_signature_def', + 'saved_model.signature_def_utils.predict_signature_def') +@deprecation.deprecated_endpoints( + 'saved_model.signature_def_utils.predict_signature_def') def predict_signature_def(inputs, outputs): """Creates prediction signature from given inputs and outputs. @@ -239,7 +252,10 @@ def _supervised_signature_def( return signature_def -@tf_export('saved_model.signature_def_utils.is_valid_signature') +@tf_export('saved_model.is_valid_signature', + 'saved_model.signature_def_utils.is_valid_signature') +@deprecation.deprecated_endpoints( + 'saved_model.signature_def_utils.is_valid_signature') def is_valid_signature(signature_def): """Determine whether a SignatureDef can be served by TensorFlow Serving.""" if signature_def is None: @@ -313,4 +329,3 @@ def _is_valid_classification_signature(signature_def): return False return True - diff --git a/tensorflow/python/saved_model/utils_impl.py b/tensorflow/python/saved_model/utils_impl.py index 06d09325c8..0bba7b6fac 100644 --- a/tensorflow/python/saved_model/utils_impl.py +++ b/tensorflow/python/saved_model/utils_impl.py @@ -27,13 +27,16 @@ from tensorflow.python.framework import sparse_tensor from tensorflow.python.lib.io import file_io from tensorflow.python.saved_model import constants from tensorflow.python.util import compat +from tensorflow.python.util import deprecation from tensorflow.python.util.tf_export import tf_export # TensorInfo helpers. -@tf_export("saved_model.utils.build_tensor_info") +@tf_export("saved_model.build_tensor_info", + "saved_model.utils.build_tensor_info") +@deprecation.deprecated_endpoints("saved_model.utils.build_tensor_info") def build_tensor_info(tensor): """Utility function to build TensorInfo proto. @@ -57,7 +60,10 @@ def build_tensor_info(tensor): return tensor_info -@tf_export("saved_model.utils.get_tensor_from_tensor_info") +@tf_export("saved_model.get_tensor_from_tensor_info", + "saved_model.utils.get_tensor_from_tensor_info") +@deprecation.deprecated_endpoints( + "saved_model.utils.get_tensor_from_tensor_info") def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): """Returns the Tensor or SparseTensor described by a TensorInfo proto. |