Caffe2 - C++ API
A deep learning, cross platform ML framework
memonger.h
1 
17 #ifndef CAFFE2_CORE_MEMONGER_H_
18 #define CAFFE2_CORE_MEMONGER_H_
19 
20 #include <unordered_set>
21 
22 #include "caffe2/core/common.h"
23 #include "caffe2/core/workspace.h"
24 #include "caffe2/proto/caffe2.pb.h"
25 
26 namespace caffe2 {
27 namespace memonger {
28 
29 NetDef optimize_inference_net(
30  const NetDef& net,
31  const std::set<string>& static_blobs);
32 
33 NetDef compute_blob_recycling_for_dag(
34  const NetDef& net,
35  const std::vector<string>& heads,
36  const std::vector<int>& op_indices,
37  const std::unordered_set<string>& shareable_blob_names,
38  const string& namescope,
39  const std::unordered_set<string>& dont_share_blob_names,
40  const std::unordered_map<string, vector<int>>& blob_shapes);
41 
42 } // memonger
43 } // caffe2
44 
45 #endif
Copyright (c) 2016-present, Facebook, Inc.