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

Return NOT_FOUND error if we cannot find a device for a kernel.

Browse Source 
PiperOrigin-RevId: 257099659
This commit is contained in:
Xiao Yu 2019-07-08 18:32:28 -07:00 committed by TensorFlower Gardener
parent 4c131dc3f7
commit bdfa6ed2d2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -243,7 +243,7 @@ Status SelectDevice(EagerOperation* op, const NodeDef& ndef, EagerContext* ctx,
TF_RETURN_IF_ERROR(SupportedDeviceTypesForNode( TF_RETURN_IF_ERROR(SupportedDeviceTypesForNode(
ctx->prioritized_device_type_list(), ndef, &supported_devs)); ctx->prioritized_device_type_list(), ndef, &supported_devs));
if (supported_devs.empty()) { if (supported_devs.empty()) {
return errors::Internal("Could not find valid device for node.\nNode:", return errors::NotFound("Could not find valid device for node.\nNode:",
FormatNodeDefForError(ndef), FormatNodeDefForError(ndef),
"\nAll kernels registered for op ", ndef.op(), "\nAll kernels registered for op ", ndef.op(),
" :\n", KernelsRegisteredForOp(ndef.op())); " :\n", KernelsRegisteredForOp(ndef.op()));