diff options
author | Kiril Gorovoy <kgorovoy@google.com> | 2016-11-16 17:02:40 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-16 17:08:56 -0800 |
commit | 0b5738774238baa832133983184da32dd2aa985a (patch) | |
tree | 9e9ed3a35fdae9e49726b2c354ab65ed9bd8ffa6 /tensorflow/contrib/session_bundle/bundle_shim.cc | |
parent | 4919aee13732911ce9214360a3740e62ea18db02 (diff) |
Change visibility of the SavedModel shim to public.
Delete bundle_shim_constants.h and use the new constant in signature_constants.h instead.
Change: 139399882
Diffstat (limited to 'tensorflow/contrib/session_bundle/bundle_shim.cc')
-rw-r--r-- | tensorflow/contrib/session_bundle/bundle_shim.cc | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/contrib/session_bundle/bundle_shim.cc b/tensorflow/contrib/session_bundle/bundle_shim.cc index 5770719cdc..a63fc77c76 100644 --- a/tensorflow/contrib/session_bundle/bundle_shim.cc +++ b/tensorflow/contrib/session_bundle/bundle_shim.cc @@ -17,7 +17,6 @@ limitations under the License. #include "tensorflow/cc/saved_model/loader.h" #include "tensorflow/cc/saved_model/signature_constants.h" -#include "tensorflow/contrib/session_bundle/bundle_shim_constants.h" #include "tensorflow/contrib/session_bundle/manifest.pb.h" #include "tensorflow/contrib/session_bundle/session_bundle.h" #include "tensorflow/contrib/session_bundle/signature.h" @@ -62,7 +61,7 @@ Status ConvertDefaultSignatureToSignatureDef(const Signatures& signatures, kRegressInputs, &signature_def); AddOutputToSignatureDef(regression_signature.output().tensor_name(), kRegressOutputs, &signature_def); - (*meta_graph_def->mutable_signature_def())[kDefaultSignatureDefKey] = + (*meta_graph_def->mutable_signature_def())[kDefaultServingSignatureDefKey] = signature_def; return Status::OK(); } else if (default_signature.type_case() == @@ -77,7 +76,7 @@ Status ConvertDefaultSignatureToSignatureDef(const Signatures& signatures, kClassifyOutputClasses, &signature_def); AddOutputToSignatureDef(classification_signature.scores().tensor_name(), kClassifyOutputScores, &signature_def); - (*meta_graph_def->mutable_signature_def())[kDefaultSignatureDefKey] = + (*meta_graph_def->mutable_signature_def())[kDefaultServingSignatureDefKey] = signature_def; return Status::OK(); } @@ -124,7 +123,7 @@ Status ConvertNamedSignaturesToSignatureDef(const Signatures& signatures, } // Add the `default` key to the signature def map of the meta graph def and // map it to the constructed signature def. - (*meta_graph_def->mutable_signature_def())[kDefaultSignatureDefKey] = + (*meta_graph_def->mutable_signature_def())[kDefaultServingSignatureDefKey] = signature_def; return Status::OK(); } |