projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e07e70a) | patch
[tf.data] Add `tf.contrib.data.prefetch_to_device()`, which supports prefetching...
authorDerek Murray <mrry@google.com>
Fri, 23 Mar 2018 01:20:09 +0000 (18:20 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 23 Mar 2018 01:22:46 +0000 (18:22 -0700)
commitdbea93d7f14be9451f1ec27c0b990fa680dee947
treedea89235bb61fa3871ef9a775e942b6210a86028tree | snapshot
parente07e70a414e83f0b15d3f7f94b9134bd86d4a1c7commit | diff
[tf.data] Add `tf.contrib.data.prefetch_to_device()`, which supports prefetching to GPU memory.

PiperOrigin-RevId: 190158272
tensorflow/contrib/data/BUILD diff | blob | history
tensorflow/contrib/data/__init__.py diff | blob | history
tensorflow/contrib/data/kernels/prefetching_kernels.cc diff | blob | history
tensorflow/contrib/data/ops/dataset_ops.cc diff | blob | history
tensorflow/contrib/data/python/kernel_tests/BUILD diff | blob | history
tensorflow/contrib/data/python/kernel_tests/prefetching_ops_test.py diff | blob | history
tensorflow/contrib/data/python/ops/BUILD diff | blob | history
tensorflow/contrib/data/python/ops/prefetching_ops.py diff | blob | history
Domain: Machine Learning / ML Framework;