| 150 | } |
| 151 | |
| 152 | Status GetRegressionSignature(const tensorflow::MetaGraphDef& meta_graph_def, |
| 153 | RegressionSignature* signature) { |
| 154 | Signatures signatures; |
| 155 | TF_RETURN_IF_ERROR(GetSignatures(meta_graph_def, &signatures)); |
| 156 | if (!signatures.has_default_signature()) { |
| 157 | return errors::FailedPrecondition( |
| 158 | strings::StrCat("Expected a default signature in: ", |
| 159 | DebugStringIfAvailable(signatures))); |
| 160 | } |
| 161 | if (!signatures.default_signature().has_regression_signature()) { |
| 162 | return errors::FailedPrecondition(strings::StrCat( |
| 163 | "Expected a regression signature in: ", |
| 164 | DebugStringIfAvailable(signatures.default_signature()))); |
| 165 | } |
| 166 | *signature = signatures.default_signature().regression_signature(); |
| 167 | return Status::OK(); |
| 168 | } |
| 169 | |
| 170 | Status RunRegression(const RegressionSignature& signature, |
| 171 | const Tensor& regression_input, Session* session, |