aboutsummaryrefslogtreecommitdiffhomepage
diff options
context:
space:
mode:
-rw-r--r--tensorflow/contrib/tpu/__init__.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/tpu/__init__.py b/tensorflow/contrib/tpu/__init__.py
index bb60f3e2d7..dc90668559 100644
--- a/tensorflow/contrib/tpu/__init__.py
+++ b/tensorflow/contrib/tpu/__init__.py
@@ -43,6 +43,7 @@
@@TPUEstimator
@@TPUEstimatorSpec
@@RunConfig
+@@InputPipelineConfig
@@TPUConfig
"""