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