Add a classification signature to the test saved_model_half_plus_two_model.

kirilg · tensorflower-gardener · commit 0da4acdf2257 · 2017-03-10T10:29:07.000-08:00
Test using this classification signature using the Predict API.
Change: 149769547
diff --git a/tensorflow/cc/saved_model/testdata/half_plus_two/00000123/saved_model.pb b/tensorflow/cc/saved_model/testdata/half_plus_two/00000123/saved_model.pb
diff --git a/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/saved_model.pbtxt b/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/saved_model.pbtxt
@@ -447,7 +447,7 @@ meta_graphs {
       }
     }
     tags: "serve"
-    tensorflow_version: "0.12.head"
+    tensorflow_version: "1.0.0"
     tensorflow_git_version: "unknown"
   }
   graph_def {
@@ -1714,7 +1714,7 @@ meta_graphs {
             dtype: DT_STRING
             tensor_shape {
             }
-            string_val: "_temp_aeab824a1fc94305a10a2504f5995de2/part"
+            string_val: "_temp_d286b725003942fd8bac94b6c67e7c0c/part"
           }
         }
       }
@@ -2503,6 +2503,42 @@ meta_graphs {
       }
     }
   }
+  signature_def {
+    key: "classify_x2_to_y3"
+    value {
+      inputs {
+        key: "inputs"
+        value {
+          name: "x2:0"
+          dtype: DT_FLOAT
+          tensor_shape {
+            dim {
+              size: -1
+            }
+            dim {
+              size: 1
+            }
+          }
+        }
+      }
+      outputs {
+        key: "scores"
+        value {
+          name: "y3:0"
+          dtype: DT_FLOAT
+          tensor_shape {
+            dim {
+              size: -1
+            }
+            dim {
+              size: 1
+            }
+          }
+        }
+      }
+      method_name: "tensorflow/serving/classify"
+    }
+  }
   signature_def {
     key: "classify_x_to_y"
     value {
diff --git a/tensorflow/examples/saved_model/saved_model_half_plus_two.py b/tensorflow/examples/saved_model/saved_model_half_plus_two.py
@@ -181,6 +181,8 @@ def _generate_saved_model_for_half_plus_two(export_dir, as_text=False):
                 _build_regression_signature(x2, y3),
             "classify_x_to_y":
                 _build_classification_signature(serialized_tf_example, y),
+            "classify_x2_to_y3":
+                _build_classification_signature(x2, y3),
             tf.saved_model.signature_constants.
             DEFAULT_SERVING_SIGNATURE_DEF_KEY:
                 predict_signature_def
