diff options
| author |  Kiril Gorovoy <kgorovoy@google.com> | 2016-11-30 15:01:22 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-30 15:05:48 -0800 |
| commit | bf3d832604e0471631f65b54e159e4d6e4ca5728 (patch) | |
| tree | 755839938c84df578bfa5e6408c2fef5915d3053 | |
| parent | 0f00cc781783feabb9f752336f34969c963f9d71 (diff) | |
Add a SavedModel ModelServer test.
- Update saved_model_half_plus_two to export a prediction default serving signature.
Change: 140657101
| -rw-r--r-- | tensorflow/python/saved_model/example/saved_model_half_plus_two.py | 19 |
1 files changed, 8 insertions, 11 deletions
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 index 9d01b962ac..d0b7b80674 100644 --- a/tensorflow/python/saved_model/example/saved_model_half_plus_two.py +++ b/tensorflow/python/saved_model/example/saved_model_half_plus_two.py @@ -128,18 +128,13 @@ def _generate_saved_model_for_half_plus_two(export_dir, as_text=False): # Set up the signature for Predict with input and output tensor # specification. predict_input_tensor = meta_graph_pb2.TensorInfo() - predict_input_tensor.name = "x" + predict_input_tensor.name = x.name predict_signature_inputs = { - signature_constants.PREDICT_INPUTS: predict_input_tensor - } - - predict_output_tensor = meta_graph_pb2.TensorInfo() - predict_output_tensor.name = "y" - predict_signature_outputs = { - signature_constants.PREDICT_OUTPUTS: predict_output_tensor + "x": predict_input_tensor } predict_signature_def = signature_def_utils.build_signature_def( - predict_signature_inputs, predict_signature_outputs, + {"x": predict_input_tensor}, + {"y": output_tensor}, signature_constants.PREDICT_METHOD_NAME) # Initialize all variables and then save the SavedModel. @@ -147,8 +142,10 @@ def _generate_saved_model_for_half_plus_two(export_dir, as_text=False): builder.add_meta_graph_and_variables( sess, [tag_constants.SERVING], signature_def_map={ - signature_constants.REGRESS_METHOD_NAME: signature_def, - signature_constants.PREDICT_METHOD_NAME: predict_signature_def + signature_constants.REGRESS_METHOD_NAME: + signature_def, + signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY: + predict_signature_def }, assets_collection=tf.get_collection(tf.GraphKeys.ASSET_FILEPATHS), legacy_init_op=tf.group(assign_filename_op)) |