diff options
author | 2018-05-11 17:14:29 -0700 | |
---|---|---|
committer | 2018-05-11 17:17:23 -0700 | |
commit | d43cb8d7358fecacef076fdab42dae03911edfc5 (patch) | |
tree | da5e4273fc7507b77ef190918b754fdd920d616a /tensorflow/contrib/data/__init__.py | |
parent | 2f5f2cb4253b4eaf7953cf7ed28f76e0bdee6fcc (diff) |
Add hook for checkpointing input pipeline while training with Estimator.
PiperOrigin-RevId: 196331223
Diffstat (limited to 'tensorflow/contrib/data/__init__.py')
-rw-r--r-- | tensorflow/contrib/data/__init__.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/data/__init__.py b/tensorflow/contrib/data/__init__.py index 077cbba9d2..4f2c72b660 100644 --- a/tensorflow/contrib/data/__init__.py +++ b/tensorflow/contrib/data/__init__.py @@ -72,6 +72,7 @@ from tensorflow.contrib.data.python.ops.grouping import group_by_window from tensorflow.contrib.data.python.ops.interleave_ops import parallel_interleave from tensorflow.contrib.data.python.ops.interleave_ops import sample_from_datasets from tensorflow.contrib.data.python.ops.interleave_ops import sloppy_interleave +from tensorflow.contrib.data.python.ops.iterator_ops import CheckpointInputPipelineHook from tensorflow.contrib.data.python.ops.iterator_ops import make_saveable_from_iterator from tensorflow.contrib.data.python.ops.prefetching_ops import prefetch_to_device from tensorflow.contrib.data.python.ops.readers import make_batched_features_dataset |