TensorFlow 2 version
|
All public utility methods for exporting Estimator to SavedModel.
This file includes functions and constants from core (model_utils) and export.py
Classes
class ClassificationOutput: Represents the output of a classification head.
class ExportOutput: Represents an output of a model that can be served.
class PredictOutput: Represents the output of a generic prediction head.
class RegressionOutput: Represents the output of a regression head.
class ServingInputReceiver: A return type for a serving_input_receiver_fn.
class TensorServingInputReceiver: A return type for a serving_input_receiver_fn.
Functions
build_parsing_serving_input_receiver_fn(...): Build a serving_input_receiver_fn expecting fed tf.Examples.
build_raw_serving_input_receiver_fn(...): Build a serving_input_receiver_fn expecting feature Tensors.