1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| #pragma once
|
| #include "caffe2/core/net.h"
| #include "caffe2/core/workspace.h"
| #include "caffe2/predictor/InferenceGraph.h"
|
| namespace caffe2 {
|
|
| // Make all operators of a given type inplace when possible
| void InPlaceOps(const InferenceGraph& graph_, const std::string& op_type);
|
| void RemoveOpsByType(const InferenceGraph& graph_, const std::string& op_type);
|
| } // namespace caffe2
|
|
