tf.estimator.export.build_raw_serving_input_receiver_fn

Build a serving_input_receiver_fn expecting feature Tensors.

tf.estimator.export.build_raw_serving_input_receiver_fn(
    features, default_batch_size=None
)

Creates an serving_input_receiver_fn that expects all features to be fed directly.

Args
features a dict of string to Tensor.
default_batch_size the number of query examples expected per batch. Leave unset for variable batch size (recommended).
Returns
A serving_input_receiver_fn.

© 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/estimator/export/build_raw_serving_input_receiver_fn