Bug fix in SavedModel shim's named signature up-conversion for classification signatures, where the value was not actually being set.

Change: 144505027

1 files changed, 2 insertions, 3 deletions

diff --git a/tensorflow/contrib/session_bundle/bundle_shim.cc b/tensorflow/contrib/session_bundle/bundle_shim.cc
index fd0ea9a2fb..9c7cdf192d 100644
--- a/tensorflow/contrib/session_bundle/bundle_shim.cc
+++ b/tensorflow/contrib/session_bundle/bundle_shim.cc
@@ -197,14 +197,13 @@ Status ConvertNamedSignaturesToSignatureDef(const Signatures& signatures,
         continue;
       }
     }
-    SignatureDef signature_def;
     const Signature signature = it_named_signature.second;
     if (IsRegressionSignature(signature)) {
       (*meta_graph_def->mutable_signature_def())[key] =
           BuildRegressionSignatureDef(signature.regression_signature());
     } else if (IsClassificationSignature(signature)) {
-      (*meta_graph_def->mutable_signature_def())[key] = signature_def;
-      BuildClassificationSignatureDef(signature.classification_signature());
+      (*meta_graph_def->mutable_signature_def())[key] =
+          BuildClassificationSignatureDef(signature.classification_signature());
     } else {
       LOG(WARNING)
           << "Named signature up-conversion to SignatureDef is only supported "