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

Making PrefetchToDevice work on XLA Compile on Demand mode. Also adds a bunch... 

Making PrefetchToDevice work on XLA Compile on Demand mode. Also adds a bunch of dataset / iterator kernel registrations for XLA.

PiperOrigin-RevId: 207802858
parent 6f07c3f3
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