mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +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
34 lines
817 B
C++
34 lines
817 B
C++
#ifndef CAFFE2_CORE_MEMONGER_H_
|
|
#define CAFFE2_CORE_MEMONGER_H_
|
|
|
|
#include <unordered_set>
|
|
|
|
#include "caffe2/core/common.h"
|
|
#include "caffe2/core/workspace.h"
|
|
#include "caffe2/proto/caffe2_pb.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
// op schema check
|
|
TORCH_API void run_schema_check(const NetDef& net);
|
|
|
|
namespace memonger {
|
|
|
|
TORCH_API NetDef optimize_inference_net(
|
|
const NetDef& net,
|
|
const std::set<string>& static_blobs);
|
|
|
|
TORCH_API NetDef compute_blob_recycling_for_dag(
|
|
const NetDef& net,
|
|
const std::vector<string>& heads,
|
|
const std::vector<int>& op_indices,
|
|
const std::unordered_set<string>& shareable_blob_names,
|
|
const string& namescope,
|
|
const std::unordered_set<string>& dont_share_blob_names,
|
|
const std::unordered_map<string, vector<int>>& blob_shapes);
|
|
|
|
} // memonger
|
|
} // caffe2
|
|
|
|
#endif
|