Stay organized with collections Save and categorize content based on your preferences.

TensorFlow 2 version View source on GitHub

Build a serving_input_receiver_fn expecting fed tf.Examples.

Creates a serving_input_receiver_fn that expects a serialized tf.Example fed into a string placeholder. The function parses the tf.Example according to the provided feature_spec, and returns all parsed Tensors as features.

feature_spec a dict of string to VarLenFeature/FixedLenFeature.
default_batch_size the number of query examples expected per batch. Leave unset for variable batch size (recommended).

A serving_input_receiver_fn suitable for use in serving.