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/cmake | |
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/cmake')
-rw-r--r-- | tensorflow/contrib/cmake/python_modules.txt | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/python_modules.txt b/tensorflow/contrib/cmake/python_modules.txt index a5eba5a8c9..40041d9c88 100644 --- a/tensorflow/contrib/cmake/python_modules.txt +++ b/tensorflow/contrib/cmake/python_modules.txt @@ -14,6 +14,7 @@ tensorflow/examples/tutorials tensorflow/examples/tutorials/mnist tensorflow/python tensorflow/python/client +tensorflow/python/compat tensorflow/python/data tensorflow/python/data/ops tensorflow/python/data/util |