commit b915de19fe04aab3d91f37547867c946290693c1
author Haoyu Zhang <haoyuzhang@google.com> Mon Oct 19 15:19:34 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Oct 19 15:34:53 2020 -0700
tree bed0c10ff5d405db3e23b6de2de588dd2bf38279
parent 06384d97dfd1c033222a71aec43c837360c12bc0

Preserve error type from EagerCopyToDevice.

PiperOrigin-RevId: 337945081
Change-Id: Ia7109d4e1532db489d8914309267928867d73036

tensorflow/core/common_runtime/eager/execute.cc

diff --git a/tensorflow/core/common_runtime/eager/execute.cc b/tensorflow/core/common_runtime/eager/execute.cc
index 5102681..572615b 100644
--- a/tensorflow/core/common_runtime/eager/execute.cc
+++ b/tensorflow/core/common_runtime/eager/execute.cc
@@ -172,10 +172,11 @@
                         /* mirror= */ true, &result_handle);
   activity.Stop();
   if (!status.ok()) {
-    return errors::Internal("Failed copying input tensor from ",
-                            handle_device->name(), " to ",
-                            expected_input_device->name(), " in order to run ",
-                            op->Name(), ": ", status.error_message());
+    return Status(
+        status.code(),
+        absl::StrCat("Failed copying input tensor from ", handle_device->name(),
+                     " to ", expected_input_device->name(), " in order to run ",
+                     op->Name(), ": ", status.error_message()));
   }
 
   *result = result_handle;