diff options
author | Sukriti Ramesh <sukritiramesh@google.com> | 2016-09-21 08:32:45 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-21 09:48:48 -0700 |
commit | 939309de3ad82c70832fc8b483e92587e91fd6fc (patch) | |
tree | 3e91986f0792a0ecc9db7ac25ee283ab5a020d2f /tensorflow/cc/saved_model/loader.h | |
parent | 9717135c4449b49fc937c597a538bd9ea3fe7412 (diff) |
Initial version of SavedModel loader in C++.
Change: 133839760
Diffstat (limited to 'tensorflow/cc/saved_model/loader.h')
-rw-r--r-- | tensorflow/cc/saved_model/loader.h | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/tensorflow/cc/saved_model/loader.h b/tensorflow/cc/saved_model/loader.h new file mode 100644 index 0000000000..3c49583eb5 --- /dev/null +++ b/tensorflow/cc/saved_model/loader.h @@ -0,0 +1,48 @@ +/* Copyright 2016 Google Inc. 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. +==============================================================================*/ + +// SavedModel loading functions and SavedModelBundle struct. + +#ifndef THIRD_PARTY_TENSORFLOW_CC_SAVED_MODEL_LOADER_H_ +#define THIRD_PARTY_TENSORFLOW_CC_SAVED_MODEL_LOADER_H_ + +#include <string> +#include <unordered_set> + +#include "tensorflow/core/lib/core/status.h" +#include "tensorflow/core/protobuf/meta_graph.pb.h" +#include "tensorflow/core/public/session.h" + +namespace tensorflow { + +// SavedModel representation once the SavedModel is loaded from storage. +struct SavedModelBundle { + std::unique_ptr<Session> session; + MetaGraphDef meta_graph_def; +}; + +// Loads a SavedModel from the specified export directory. The meta graph def to +// be loaded is identified by the supplied tags, corresponding exactly to the +// set of tags used at SavedModel build time. Returns a SavedModel bundle with a +// session and the requested meta graph def, if found. +Status LoadSavedModel(const string& export_dir, + const std::unordered_set<string>& tags, + const SessionOptions& session_options, + const RunOptions& run_options, + SavedModelBundle* const bundle); + +} // namespace tensorflow + +#endif // THIRD_PARTY_TENSORFLOW_CC_SAVED_MODEL_LOADER_H_ |