| 198 | } |
| 199 | |
| 200 | Status GetGenericSignature(const string& name, |
| 201 | const tensorflow::MetaGraphDef& meta_graph_def, |
| 202 | GenericSignature* signature) { |
| 203 | Signatures signatures; |
| 204 | TF_RETURN_IF_ERROR(GetSignatures(meta_graph_def, &signatures)); |
| 205 | const auto& it = signatures.named_signatures().find(name); |
| 206 | if (it == signatures.named_signatures().end()) { |
| 207 | return errors::InvalidArgument( |
| 208 | strings::StrCat("Missing generic signature named \"", name, "\" in ", |
| 209 | DebugStringIfAvailable(signatures))); |
| 210 | } |
| 211 | if (!it->second.has_generic_signature()) { |
| 212 | return errors::InvalidArgument(strings::StrCat( |
| 213 | "Expected a generic signature: ", DebugStringIfAvailable(it->second))); |
| 214 | } |
| 215 | *signature = it->second.generic_signature(); |
| 216 | return Status::OK(); |
| 217 | } |
| 218 | |
| 219 | Status GetDefaultSignature(const tensorflow::MetaGraphDef& meta_graph_def, |
| 220 | Signature* default_signature) { |