From bd221538223827a96377758a3f8365c93ee64d45 Mon Sep 17 00:00:00 2001 From: Rohan Jain Date: Fri, 6 Jul 2018 16:21:01 -0700 Subject: 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 --- tensorflow/contrib/data/__init__.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'tensorflow/contrib/data/__init__.py') 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 -- cgit v1.2.3