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

Creating a wrapper around DatasetVariantWrapper's that can be passed from CPU 

to GPU while keeping everything in host memory. This is necessary for when we
move to creating datasets on the go and for transformations such as
copy_to_device we need to ferry the DatasetVariantWrapper.

PiperOrigin-RevId: 224531292
parent 8d3be0f5
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