Add signature constants for SavedModel in py.

sukritiramesh · tensorflower-gardener · commit f5c19e512c17 · 2016-10-10T10:48:24.000-07:00
Change: 135691893
diff --git a/tensorflow/python/saved_model/BUILD b/tensorflow/python/saved_model/BUILD
@@ -15,6 +15,12 @@ py_library(
     srcs_version = "PY2AND3",
 )
 
+py_library(
+    name = "signature_constants",
+    srcs = ["signature_constants.py"],
+    srcs_version = "PY2AND3",
+)
+
 py_library(
     name = "builder",
     srcs = ["builder.py"],
diff --git a/tensorflow/python/saved_model/example/BUILD b/tensorflow/python/saved_model/example/BUILD
@@ -35,6 +35,7 @@ py_binary(
         "//tensorflow/core:protos_all_py",
         "//tensorflow/python/saved_model:builder",
         "//tensorflow/python/saved_model:constants",
+        "//tensorflow/python/saved_model:signature_constants",
         "//tensorflow/python/saved_model:utils",
     ],
 )
diff --git a/tensorflow/python/saved_model/example/saved_model_half_plus_two.py b/tensorflow/python/saved_model/example/saved_model_half_plus_two.py
@@ -34,6 +34,7 @@
 from tensorflow.core.protobuf import meta_graph_pb2
 from tensorflow.python.saved_model import builder as saved_model_builder
 from tensorflow.python.saved_model import constants
+from tensorflow.python.saved_model import signature_constants
 from tensorflow.python.saved_model import utils
 
 
@@ -67,19 +68,23 @@ def _generate_saved_model_for_half_plus_two(export_dir, as_text=False):
     # specification.
     input_tensor = meta_graph_pb2.TensorInfo()
     input_tensor.name = serialized_tf_example.name
-    signature_inputs = {"input": input_tensor}
+    signature_inputs = {signature_constants.REGRESS_INPUTS: input_tensor}
 
     output_tensor = meta_graph_pb2.TensorInfo()
     output_tensor.name = tf.identity(y).name
-    signature_outputs = {"output": output_tensor}
-    signature_def = utils.build_signature_def(signature_inputs,
-                                              signature_outputs, "regression")
+    signature_outputs = {signature_constants.REGRESS_OUTPUTS: output_tensor}
+    signature_def = utils.build_signature_def(
+        signature_inputs, signature_outputs,
+        signature_constants.REGRESS_METHOD_NAME)
 
     # Initialize all variables and then save the SavedModel.
     sess.run(tf.initialize_all_variables())
     builder.add_meta_graph_and_variables(
         sess, [constants.TAG_SERVING],
-        signature_def_map={"regression": signature_def})
+        signature_def_map={
+            signature_constants.REGRESS_METHOD_NAME:
+                signature_def
+        })
     builder.save(as_text)
 
 
diff --git a/tensorflow/python/saved_model/signature_constants.py b/tensorflow/python/saved_model/signature_constants.py
@@ -0,0 +1,61 @@
+# Copyright 2016 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+"""Signature constants for SavedModel save and restore operations.
+
+"""
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
+################################################################################
+# Classification API constants.
+
+# Classification inputs.
+CLASSIFY_INPUTS = "inputs"
+
+# Classification method name used in a SignatureDef.
+CLASSIFY_METHOD_NAME = "tensorflow/serving/classify"
+
+# Classification classes output.
+CLASSIFY_OUTPUT_CLASSES = "classes"
+
+# Classification scores output.
+CLASSIFY_OUTPUT_SCORES = "scores"
+
+################################################################################
+# Prediction API constants.
+
+# Predict inputs.
+PREDICT_INPUTS = "inputs"
+
+# Prediction method name used in a SignatureDef.
+PREDICT_METHOD_NAME = "tensorflow/serving/predict"
+
+# Predict outputs.
+PREDICT_OUTPUTS = "outputs"
+
+################################################################################
+# Regression API constants.
+
+# Regression inputs.
+REGRESS_INPUTS = "inputs"
+
+# Regression method name used in a SignatureDef.
+REGRESS_METHOD_NAME = "tensorflow/serving/regress"
+
+# Regression outputs.
+REGRESS_OUTPUTS = "outputs"
+
+################################################################################

Original file line number	Diff line number	Diff line change
`@@ -35,6 +35,7 @@ py_binary(`
`35`	`35`	`"//tensorflow/core:protos_all_py",`
`36`	`36`	`"//tensorflow/python/saved_model:builder",`
`37`	`37`	`"//tensorflow/python/saved_model:constants",`
	`38`	`+ "//tensorflow/python/saved_model:signature_constants",`
`38`	`39`	`"//tensorflow/python/saved_model:utils",`
`39`	`40`	`],`
`40`	`41`	`)`