diff options
| author |  Derek Murray <mrry@google.com> | 2018-03-22 18:20:09 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-22 18:22:46 -0700 |
| commit | dbea93d7f14be9451f1ec27c0b990fa680dee947 (patch) | |
| tree | dea89235bb61fa3871ef9a775e942b6210a86028 /tensorflow/contrib/data/__init__.py | |
| parent | e07e70a414e83f0b15d3f7f94b9134bd86d4a1c7 (diff) | |
[tf.data] Add `tf.contrib.data.prefetch_to_device()`, which supports prefetching to GPU memory.
PiperOrigin-RevId: 190158272
Diffstat (limited to 'tensorflow/contrib/data/__init__.py')
| -rw-r--r-- | tensorflow/contrib/data/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/data/__init__.py b/tensorflow/contrib/data/__init__.py index 9212b69700..766721d8d2 100644 --- a/tensorflow/contrib/data/__init__.py +++ b/tensorflow/contrib/data/__init__.py @@ -36,6 +36,7 @@ See the @{$datasets$Importing Data} Programmer's Guide for an overview. @@map_and_batch @@padded_batch_and_drop_remainder @@parallel_interleave +@@prefetch_to_device @@read_batch_features @@rejection_resample @@scan @@ -67,6 +68,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 sloppy_interleave 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 from tensorflow.contrib.data.python.ops.readers import read_batch_features from tensorflow.contrib.data.python.ops.readers import SqlDataset |