tf.saved_model.predict_signature_def

View source on GitHub

Creates prediction signature from given inputs and outputs.

Aliases:

• tf.compat.v1.saved_model.predict_signature_def
• tf.compat.v1.saved_model.signature_def_utils.predict_signature_def
• tf.saved_model.signature_def_utils.predict_signature_def

tf.saved_model.predict_signature_def(
    inputs,
    outputs
)

This function produces signatures intended for use with the TensorFlow Serving Predict API (tensorflow_serving/apis/prediction_service.proto). This API imposes no constraints on the input and output types.

Args:

• inputs: dict of string to Tensor.
• outputs: dict of string to Tensor.

Returns:

A prediction-flavored signature_def.

Raises:

• ValueError: If inputs or outputs is None.