Public API for tf.saved_model namespace.
contains_saved_model(...)
: Checks whether the provided export directory could contain a SavedModel.
load(...)
: Load a SavedModel from export_dir
.
save(...)
: Exports the Trackable object obj
to SavedModel format.
ASSETS_DIRECTORY = 'assets'
ASSETS_KEY = 'saved_model_assets'
CLASSIFY_INPUTS = 'inputs'
CLASSIFY_METHOD_NAME = 'tensorflow/serving/classify'
CLASSIFY_OUTPUT_CLASSES = 'classes'
CLASSIFY_OUTPUT_SCORES = 'scores'
DEFAULT_SERVING_SIGNATURE_DEF_KEY = 'serving_default'
GPU = 'gpu'
PREDICT_INPUTS = 'inputs'
PREDICT_METHOD_NAME = 'tensorflow/serving/predict'
PREDICT_OUTPUTS = 'outputs'
REGRESS_INPUTS = 'inputs'
REGRESS_METHOD_NAME = 'tensorflow/serving/regress'
REGRESS_OUTPUTS = 'outputs'
SAVED_MODEL_FILENAME_PB = 'saved_model.pb'
SAVED_MODEL_FILENAME_PBTXT = 'saved_model.pbtxt'
SAVED_MODEL_SCHEMA_VERSION = 1
SERVING = 'serve'
TPU = 'tpu'
TRAINING = 'train'
VARIABLES_DIRECTORY = 'variables'
VARIABLES_FILENAME = 'variables'
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/compat/v2/saved_model