Commit 10448884 authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Making the Eager iterator use the new copy_to_device. 

This CL gets rid of the forced placement of all eager datasets / iterators on the CPU since now we can have some datasets on the GPU.

PiperOrigin-RevId: 205264791
parent 9dfa333c
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment