| 26 | namespace swig { |
| 27 | |
| 28 | string TryFindKernelClass(const string& serialized_node_def) { |
| 29 | tensorflow::NodeDef node_def; |
| 30 | if (!node_def.ParseFromString(serialized_node_def)) { |
| 31 | LOG(WARNING) << "Error parsing node_def"; |
| 32 | return ""; |
| 33 | } |
| 34 | |
| 35 | const tensorflow::OpRegistrationData* op_reg_data; |
| 36 | auto status = |
| 37 | tensorflow::OpRegistry::Global()->LookUp(node_def.op(), &op_reg_data); |
| 38 | if (!status.ok()) { |
| 39 | LOG(WARNING) << "Op " << node_def.op() << " not found: " << status; |
| 40 | return ""; |
| 41 | } |
| 42 | AddDefaultsToNodeDef(op_reg_data->op_def, &node_def); |
| 43 | |
| 44 | tensorflow::DeviceNameUtils::ParsedName parsed_name; |
| 45 | if (!tensorflow::DeviceNameUtils::ParseFullName(node_def.device(), |
| 46 | &parsed_name)) { |
| 47 | LOG(WARNING) << "Failed to parse device from node_def: " |
| 48 | << node_def.ShortDebugString(); |
| 49 | return ""; |
| 50 | } |
| 51 | string class_name = ""; |
| 52 | tensorflow::FindKernelDef(tensorflow::DeviceType(parsed_name.type.c_str()), |
| 53 | node_def, nullptr /* kernel_def */, &class_name) |
| 54 | .IgnoreError(); |
| 55 | return class_name; |
| 56 | } |
| 57 | |
| 58 | } // namespace swig |
| 59 | } // namespace tensorflow |
nothing calls this directly
no test coverage detected