W3cubDocs

/TensorFlow 2.4

Module: tf.compat.v1.saved_model

Public API for tf.saved_model namespace.

Modules

builder module: SavedModel builder.

constants module: Constants for SavedModel save and restore operations.

experimental module: Public API for tf.saved_model.experimental namespace.

loader module: Loader functionality for SavedModel with hermetic, language-neutral exports.

main_op module: SavedModel main op.

signature_constants module: Signature constants for SavedModel save and restore operations.

signature_def_utils module: SignatureDef utility functions.

tag_constants module: Common tags used for graphs in SavedModel.

utils module: SavedModel utility functions.

Classes

class Asset: Represents a file asset to hermetically include in a SavedModel.

class Builder: Builds the SavedModel protocol buffer and saves variables and assets.

class SaveOptions: Options for saving to SavedModel.

Functions

build_signature_def(...): Utility function to build a SignatureDef protocol buffer.

build_tensor_info(...): Utility function to build TensorInfo proto from a Tensor. (deprecated)

classification_signature_def(...): Creates classification signature from given examples and predictions.

contains_saved_model(...): Checks whether the provided export directory could contain a SavedModel.

get_tensor_from_tensor_info(...): Returns the Tensor or CompositeTensor described by a TensorInfo proto. (deprecated)

is_valid_signature(...): Determine whether a SignatureDef can be served by TensorFlow Serving.

load(...): Loads the model from a SavedModel as specified by tags. (deprecated)

load_v2(...): Load a SavedModel from export_dir.

main_op_with_restore(...): Returns a main op to init variables, tables and restore the graph. (deprecated)

maybe_saved_model_directory(...): Checks whether the provided export directory could contain a SavedModel.

predict_signature_def(...): Creates prediction signature from given inputs and outputs.

regression_signature_def(...): Creates regression signature from given examples and predictions.

save(...): Exports the Trackable object obj to SavedModel format.

simple_save(...): Convenience function to build a SavedModel suitable for serving. (deprecated)

Other Members
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'
DEBUG_DIRECTORY 'debug'
DEBUG_INFO_FILENAME_PB 'saved_model_debug_info.pb'
DEFAULT_SERVING_SIGNATURE_DEF_KEY 'serving_default'
GPU 'gpu'
LEGACY_INIT_OP_KEY 'legacy_init_op'
MAIN_OP_KEY 'saved_model_main_op'
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/r2.4/api_docs/python/tf/compat/v1/saved_model