projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 218647d) | patch
[tf.data] Enable using `tf.contrib.data.prefetch_to_device()` in eager mode.
authorDerek Murray <mrry@google.com>
Fri, 6 Apr 2018 16:26:26 +0000 (09:26 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 6 Apr 2018 16:31:14 +0000 (09:31 -0700)
commiteef7771ddb9fc0de5e8c9aeabcaa186ff78ec105
tree05bd9b49e51ca14b321b2897daf9b98289486559tree | snapshot
parent218647db25d1e754ad85fd1fa8a0960b82ae83bfcommit | diff
[tf.data] Enable using `tf.contrib.data.prefetch_to_device()` in eager mode.

The added functionality is a substitute for the implicit prefetching in
`tfe.Iterator`, and the two paths will converge in a future change.

Fixes #18260.

PiperOrigin-RevId: 191897666
tensorflow/contrib/data/python/ops/prefetching_ops.py diff | blob | history
tensorflow/contrib/eager/python/datasets.py diff | blob | history
tensorflow/contrib/eager/python/datasets_test.py diff | blob | history
Domain: Machine Learning / ML Framework;