W3cubDocs

/TensorFlow Python

tf.estimator.export.build_raw_serving_input_receiver_fn

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

Defined in tensorflow/python/estimator/export/export.py.

Build a serving_input_receiver_fn expecting feature Tensors.

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.

© 2018 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/api_docs/python/tf/estimator/export/build_raw_serving_input_receiver_fn