tf.contrib.tpu.export_estimator_savedmodel(
estimator,
export_dir_base,
serving_input_receiver_fn,
assets_extra=None,
as_text=False,
checkpoint_path=None,
strip_default_attrs=False
)
Defined in tensorflow/contrib/tpu/python/tpu/tpu_estimator.py
.
Export Estimator
trained model for TPU inference.
Args:
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 aServingInputReceiver
orTensorServingInputReceiver
.assets_extra
: A dict specifying how to populate the assets.extra directory within the exported SavedModel, orNone
if no extra assets are needed.as_text
: whether to write the SavedModel proto in text format.checkpoint_path
: The checkpoint path to export. IfNone
(the default), the most recent checkpoint found within the model directory is chosen.strip_default_attrs
: Boolean. IfTrue
, default-valued attributes will be removed from the NodeDefs.
Returns:
The string path to the exported directory.