experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Preserve error type from EagerCopyToDevice.

Browse Source 
PiperOrigin-RevId: 337945081
Change-Id: Ia7109d4e1532db489d8914309267928867d73036
This commit is contained in:
Haoyu Zhang 2020-10-19 15:19:34 -07:00 committed by TensorFlower Gardener
parent 06384d97df
commit b915de19fe
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/core/common_runtime/eager/execute.cc
View File

@ -172,10 +172,11 @@ Status CopyInputToExpectedDevice(EagerContext* ctx, EagerOperation* op,
/* mirror= */ true, &result_handle); /* mirror= */ true, &result_handle);
activity.Stop(); activity.Stop();
if (!status.ok()) { if (!status.ok()) {
return errors::Internal("Failed copying input tensor from ", return Status(
handle_device->name(), " to ", status.code(),
expected_input_device->name(), " in order to run ", absl::StrCat("Failed copying input tensor from ", handle_device->name(),
op->Name(), ": ", status.error_message()); " to ", expected_input_device->name(), " in order to run ",
op->Name(), ": ", status.error_message()));
} }
*result = result_handle; *result = result_handle;