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

Log a warning instead of silently ignoring errors during kernel lookup.

Browse Source 
PiperOrigin-RevId: 332928223
Change-Id: Ia8bbb34d196799047ab9969d9eb67e13d70b72bd
This commit is contained in:
Zachary Garrett 2020-09-21 14:01:08 -07:00 committed by TensorFlower Gardener
parent 906fac3188
commit 317cf7280e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/util/kernel_registry.cc
View File

@ -49,9 +49,12 @@ string TryFindKernelClass(const string& serialized_node_def) {
return ""; return "";
} }
string class_name = ""; string class_name = "";
tensorflow::FindKernelDef(tensorflow::DeviceType(parsed_name.type.c_str()), status = tensorflow::FindKernelDef(
node_def, nullptr /* kernel_def */, &class_name) tensorflow::DeviceType(parsed_name.type.c_str()), node_def,
.IgnoreError(); nullptr /* kernel_def */, &class_name);
if (!status.ok()) {
LOG(WARNING) << "Op [" << node_def.op() << "]: " << status;
}
return class_name; return class_name;
} }