Apply to speak at TensorFlow World. Deadline April 23rd. Propose talk

tf.contrib.tpu.export_estimator_savedmodel

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 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.
  • strip_default_attrs: Boolean. If True, default-valued attributes will be removed from the NodeDefs.

Returns:

The string path to the exported directory.