commit 2f0ac02d72483b109751fbe2ff170dc67fcc6216
author Cesar Crusius <ccrusius@google.com> Mon Apr 06 14:58:59 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Apr 06 15:02:17 2020 -0700
tree 077335716d634db785c27a81ea27c2adbc6f7999
parent 444d8d7cf409b2672cb16f9590b4d168deb1e059

Use canonical CPU device name in TensorHandle::Resolve.

PiperOrigin-RevId: 305124817
Change-Id: I2e51fa917503d4deab90ed57b1407f77a89c280f

tensorflow/core/common_runtime/eager/core.cc

diff --git a/tensorflow/core/common_runtime/eager/core.cc b/tensorflow/core/common_runtime/eager/core.cc
index 20500ac..de7e747 100644
--- a/tensorflow/core/common_runtime/eager/core.cc
+++ b/tensorflow/core/common_runtime/eager/core.cc
@@ -41,7 +41,7 @@
     auto* custom_device = absl::get<CustomDevice*>(device());
     TensorHandle* copy;
     *status = custom_device->CopyTensorFromDevice(
-        this, "/job:localhost/task:0/replica:0/device:CPU:0", &copy);
+        this, "/job:localhost/replica:0/task:0/device:CPU:0", &copy);
     if (status->ok()) {
       return copy->Resolve(status);
     } else {