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

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

@ -172,10 +172,11 @@ Status CopyInputToExpectedDevice(EagerContext* ctx, EagerOperation* op,
/* 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;