aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tpu/__init__.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-13 21:31:49 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-13 21:34:38 -0700
commit88b656acd480f6956894e3bb8c8f0c52fe033bc4 (patch)
tree2279825fa37fa32ce1278d78e357eff7f74e4817 /tensorflow/contrib/tpu/__init__.py
parent274cce5990ff279e2ce293ae9ce1c1d0445d3242 (diff)
Add wrapper to help export model trained with estimator as SavedModel for TPU.
PiperOrigin-RevId: 204568222
Diffstat (limited to 'tensorflow/contrib/tpu/__init__.py')
-rw-r--r--tensorflow/contrib/tpu/__init__.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tpu/__init__.py b/tensorflow/contrib/tpu/__init__.py
index d62338680e..d5484e9032 100644
--- a/tensorflow/contrib/tpu/__init__.py
+++ b/tensorflow/contrib/tpu/__init__.py
@@ -42,10 +42,10 @@
@@TPUEstimator
@@TPUEstimatorSpec
+@@export_estimator_savedmodel
@@RunConfig
@@InputPipelineConfig
@@TPUConfig
-
@@bfloat16_scope
"""