mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +01:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
28 lines
802 B
C++
28 lines
802 B
C++
#pragma once
|
|
|
|
#include "caffe2/core/db.h"
|
|
#include "caffe2/core/workspace.h"
|
|
#include "caffe2/predictor/predictor_config.h"
|
|
#include "caffe2/proto/metanet.pb.h"
|
|
|
|
namespace caffe2 {
|
|
namespace predictor_utils {
|
|
|
|
TORCH_API const NetDef& getNet(const MetaNetDef& def, const std::string& name);
|
|
const ::google::protobuf::RepeatedPtrField<::std::string>& getBlobs(
|
|
const MetaNetDef& def,
|
|
const std::string& name);
|
|
|
|
TORCH_API std::unique_ptr<MetaNetDef> extractMetaNetDef(
|
|
db::Cursor* cursor,
|
|
const std::string& key);
|
|
|
|
// Extract the MetaNetDef from `db`, and run the global init net on the
|
|
// `master` workspace.
|
|
TORCH_API std::unique_ptr<MetaNetDef> runGlobalInitialization(
|
|
std::unique_ptr<db::DBReader> db,
|
|
Workspace* master);
|
|
|
|
} // namespace predictor_utils
|
|
} // namespace caffe2
|