aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tpu/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/tpu/__init__.py')
-rw-r--r--tensorflow/contrib/tpu/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/__init__.py b/tensorflow/contrib/tpu/__init__.py
index dc90668559..d5484e9032 100644
--- a/tensorflow/contrib/tpu/__init__.py
+++ b/tensorflow/contrib/tpu/__init__.py
@@ -42,9 +42,11 @@
@@TPUEstimator
@@TPUEstimatorSpec
+@@export_estimator_savedmodel
@@RunConfig
@@InputPipelineConfig
@@TPUConfig
+@@bfloat16_scope
"""
from __future__ import absolute_import