diff --git a/tensorflow/contrib/session_bundle/manifest.proto b/tensorflow/contrib/session_bundle/manifest.proto index 499c1bcfd8ca62..482ed372dc1e5d 100644 --- a/tensorflow/contrib/session_bundle/manifest.proto +++ b/tensorflow/contrib/session_bundle/manifest.proto @@ -1,6 +1,6 @@ syntax = "proto3"; -package tensorflow.contrib; +package tensorflow.serving; // Signatures of model export. message Signatures { diff --git a/tensorflow/contrib/session_bundle/session_bundle.cc b/tensorflow/contrib/session_bundle/session_bundle.cc index d706d7638562db..5029dcd7fedb74 100644 --- a/tensorflow/contrib/session_bundle/session_bundle.cc +++ b/tensorflow/contrib/session_bundle/session_bundle.cc @@ -37,7 +37,7 @@ limitations under the License. #include "tensorflow/core/public/session_options.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace { // Create a session using the given options and load the graph. @@ -184,5 +184,5 @@ tensorflow::Status LoadSessionBundleFromPath( return Status::OK(); } -} // namespace contrib +} // namespace serving } // namespace tensorflow diff --git a/tensorflow/contrib/session_bundle/session_bundle.h b/tensorflow/contrib/session_bundle/session_bundle.h index 92ed0132b88c2b..f238ff13aecee1 100644 --- a/tensorflow/contrib/session_bundle/session_bundle.h +++ b/tensorflow/contrib/session_bundle/session_bundle.h @@ -30,7 +30,7 @@ limitations under the License. #include "tensorflow/core/public/session_options.h" namespace tensorflow { -namespace contrib { +namespace serving { const char kMetaGraphDefFilename[] = "export.meta"; const char kAssetsDirectory[] = "assets"; @@ -53,7 +53,7 @@ tensorflow::Status LoadSessionBundleFromPath( const tensorflow::SessionOptions& options, const tensorflow::StringPiece export_dir, SessionBundle* bundle); -} // namespace contrib +} // namespace serving } // namespace tensorflow #endif // THIRD_PARTY_TENSORFLOW_CONTRIB_SESSION_BUNDLE_SESSION_BUNDLE_H_ diff --git a/tensorflow/contrib/session_bundle/session_bundle_test.cc b/tensorflow/contrib/session_bundle/session_bundle_test.cc index 2bf09ad4384578..5aeb3445249d9e 100644 --- a/tensorflow/contrib/session_bundle/session_bundle_test.cc +++ b/tensorflow/contrib/session_bundle/session_bundle_test.cc @@ -33,7 +33,7 @@ limitations under the License. #include "tensorflow/core/public/session_options.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace { TEST(LoadSessionBundleFromPath, Basic) { @@ -72,7 +72,7 @@ TEST(LoadSessionBundleFromPath, Basic) { ASSERT_EQ(1, collection_def[kSignaturesKey].any_list().value_size()); collection_def[kSignaturesKey].any_list().value(0).UnpackTo(&signatures); ASSERT_TRUE(signatures.default_signature().has_regression_signature()); - const tensorflow::contrib::RegressionSignature regression_signature = + const tensorflow::serving::RegressionSignature regression_signature = signatures.default_signature().regression_signature(); const string input_name = regression_signature.input().tensor_name(); @@ -98,5 +98,5 @@ TEST(LoadSessionBundleFromPath, BadExportPath) { } } // namespace -} // namespace contrib +} // namespace serving } // namespace tensorflow diff --git a/tensorflow/contrib/session_bundle/signature.cc b/tensorflow/contrib/session_bundle/signature.cc index 3550a7d10d3621..50fea99ef2c698 100644 --- a/tensorflow/contrib/session_bundle/signature.cc +++ b/tensorflow/contrib/session_bundle/signature.cc @@ -31,7 +31,7 @@ limitations under the License. #include "tensorflow/core/public/session.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace { // Returns OK if the input and output batch sizes match. @@ -235,7 +235,7 @@ Status GetNamedSignature(const string& name, Status BindGenericInputs(const GenericSignature& signature, const std::vector>& inputs, std::vector>* bound_inputs) { - const protobuf::Map& bindings = + const protobuf::Map& bindings = signature.map(); for (const auto& entry : inputs) { @@ -252,7 +252,7 @@ Status BindGenericInputs(const GenericSignature& signature, Status BindGenericNames(const GenericSignature& signature, const std::vector& input_names, std::vector* bound_names) { - const protobuf::Map& bindings = + const protobuf::Map& bindings = signature.map(); for (const string& entry : input_names) { @@ -266,5 +266,5 @@ Status BindGenericNames(const GenericSignature& signature, return Status::OK(); } -} // namespace contrib +} // namespace serving } // namespace tensorflow diff --git a/tensorflow/contrib/session_bundle/signature.h b/tensorflow/contrib/session_bundle/signature.h index 98d6e5660126ef..2736f965dc6cb5 100644 --- a/tensorflow/contrib/session_bundle/signature.h +++ b/tensorflow/contrib/session_bundle/signature.h @@ -31,7 +31,7 @@ limitations under the License. #include "tensorflow/core/public/session.h" namespace tensorflow { -namespace contrib { +namespace serving { const char kSignaturesKey[] = "serving_signatures"; @@ -117,7 +117,7 @@ Status BindGenericInputs(const GenericSignature& signature, Status BindGenericNames(const GenericSignature& signature, const std::vector& input_names, std::vector* bound_names); -} // namespace contrib +} // namespace serving } // namespace tensorflow #endif // THIRD_PARTY_TENSORFLOW_CONTRIB_SESSION_BUNDLE_SIGNATURE_H_ diff --git a/tensorflow/contrib/session_bundle/signature_test.cc b/tensorflow/contrib/session_bundle/signature_test.cc index abeaf23c0efb8b..c89055121774c2 100644 --- a/tensorflow/contrib/session_bundle/signature_test.cc +++ b/tensorflow/contrib/session_bundle/signature_test.cc @@ -30,7 +30,7 @@ limitations under the License. #include "tensorflow/core/public/session.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace { static bool HasSubstr(const string& base, const string& substr) { @@ -598,5 +598,5 @@ TEST_F(GenericSignatureTest, BindGenericNamesMissingBinding) { } } // namespace -} // namespace contrib +} // namespace serving } // namespace tensorflow diff --git a/tensorflow/contrib/session_bundle/test_util.cc b/tensorflow/contrib/session_bundle/test_util.cc index 90870a69a04bf3..cb81a24d322e79 100644 --- a/tensorflow/contrib/session_bundle/test_util.cc +++ b/tensorflow/contrib/session_bundle/test_util.cc @@ -20,7 +20,7 @@ limitations under the License. #include "tensorflow/core/platform/test.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace test_util { string TestSrcDirPath(const string& relative_path) { @@ -31,5 +31,5 @@ string TestSrcDirPath(const string& relative_path) { } } // namespace test_util -} // namespace contrib +} // namespace serving } // namespace tensorflow diff --git a/tensorflow/contrib/session_bundle/test_util.h b/tensorflow/contrib/session_bundle/test_util.h index 87ddf28d6070fa..ca7feed4b591ab 100644 --- a/tensorflow/contrib/session_bundle/test_util.h +++ b/tensorflow/contrib/session_bundle/test_util.h @@ -23,7 +23,7 @@ limitations under the License. #include "tensorflow/core/platform/types.h" namespace tensorflow { -namespace contrib { +namespace serving { namespace test_util { // Creates an absolute test srcdir path to the linked in runfiles given a path @@ -32,7 +32,7 @@ namespace test_util { string TestSrcDirPath(const string& relative_path); } // namespace test_util -} // namespace contrib +} // namespace serving } // namespace tensorflow #endif // THIRD_PARTY_TENSORFLOW_CONTRIB_SESSION_BUNDLE_TEST_UTIL_H_