tf.contrib.tpu.export_estimator_savedmodel

View source on GitHub

Export Estimator trained model for TPU inference.

tf.contrib.tpu.export_estimator_savedmodel(
    estimator, export_dir_base, serving_input_receiver_fn, assets_extra=None,
    as_text=False, checkpoint_path=None
)

estimator Estimator with which model has been trained.
export_dir_base A string containing a directory in which to create timestamped subdirectories containing exported SavedModels.
serving_input_receiver_fn A function that takes no argument and returns a ServingInputReceiver or TensorServingInputReceiver.
assets_extra A dict specifying how to populate the assets.extra directory within the exported SavedModel, or None if no extra assets are needed.
as_text whether to write the SavedModel proto in text format.
checkpoint_path The checkpoint path to export. If None (the default), the most recent checkpoint found within the model directory is chosen.

The string path to the exported directory.