Commit 3db0e545 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

Change RecvTensor RPC implementation to use... 

Change RecvTensor RPC implementation to use DeviceContext::CopyDeviceTensorToCPU rather than calling GPUUtil::CopyGPUTensorToCPU. The direct call into the GPU code is problematic for non-GPU devices.

PiperOrigin-RevId: 195433287
parent 2d6170fc
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