Commit ebb2e9f4 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Fix OneDeviceDataset numpy dataset "Error copying tensor to device:" 

bug.  I needed to make sure OneDeviceExtended._create_variable() used
tf.device() in this case instead of tf.colocate_with(). I also needed
to remove the device set by one device strategy's scope, since
otherwise dataset creation gives lots of errors.

Also, expand OneDeviceStrategy eager/GPU test coverage.

PiperOrigin-RevId: 232605692
parent 6a8670aa
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