Commit 8d761771 authored by Akshay Modi's avatar Akshay Modi Committed by TensorFlower Gardener
Browse files

Always copy tensors to the remote CPU when executing a remote kernel. 

The copy to the appropriate device will happen when the kernel is executing.

Without this, its possible we'll copy tensors to an accelerator device when the
op expects host memory inputs.

PiperOrigin-RevId: 231696485
parent 74534b24
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