diff options
author | Rohan Jain <rohanj@google.com> | 2018-07-06 16:21:01 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-07 21:11:54 -0700 |
commit | bd221538223827a96377758a3f8365c93ee64d45 (patch) | |
tree | a60fb222bc440e48195370ad387202b78186be53 /tensorflow/contrib/data/__init__.py | |
parent | ffb9b0de1e1396617a833c08b88b8a136fd7027e (diff) |
Creating a CopyToDeviceDataset that uses GeneratorDataset and remote function calls to copy data from one device to another. Used in combination with PrefetchDataset, this can serve as a replacement for PrefetchToDeviceDataset.
This also adds GPU kernel registrations for some iterator ops and for GeneratorDataset and PrefetchDataset
PiperOrigin-RevId: 203538112
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 156538b4e0..675330716b 100644 --- a/tensorflow/contrib/data/__init__.py +++ b/tensorflow/contrib/data/__init__.py @@ -34,6 +34,7 @@ See @{$guide/datasets$Importing Data} for an overview. @@batch_and_drop_remainder @@bucket_by_sequence_length @@choose_from_datasets +@@copy_to_device @@dense_to_sparse_batch @@enumerate_dataset @@ -86,6 +87,7 @@ from tensorflow.contrib.data.python.ops.interleave_ops import sample_from_datase 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 copy_to_device from tensorflow.contrib.data.python.ops.prefetching_ops import prefetch_to_device from tensorflow.contrib.data.python.ops.random_ops import RandomDataset from tensorflow.contrib.data.python.ops.readers import CsvDataset |