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

Adding the experimental_ints_on_device attribute to function defined in... 

Adding the experimental_ints_on_device attribute to function defined in CopyToDeviceDataset. Similar to the problem with other functions that might run on a GPU, we want int32's in device memory instead because the downstream code expects it as so.

PiperOrigin-RevId: 231484444
parent e7f7052f
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