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

Creating a CopyToDeviceDataset that uses GeneratorDataset and remote function... 

Creating a CopyToDeviceDataset that uses GeneratorDataset and remote function calls to copy data from one device to another. Used in combination with PrefetchDataset, this can serve as a replacement for PrefetchToDeviceDataset.

This also adds GPU kernel registrations for some iterator ops and for GeneratorDataset and PrefetchDataset

PiperOrigin-RevId: 203538112
parent ffb9b0de
Expand all 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