| View source on GitHub |
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