|View source on GitHub|
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 )
Estimatorwith 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
assets_extra: A dict specifying how to populate the assets.extra directory within the exported SavedModel, or
Noneif 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.