Commit 491584ff authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

eager: Always run dataset iterator operations on CPU.

It has no kernels for other devices.
With an explicit "tf.device()" before invoking the kernel we ensure
that Iterator.next() functions even when placed inside a:

with tf.device("/device:GPU:0")

PiperOrigin-RevId: 171048558
parent 3b354016
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment