1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
| #ifndef CAFFE2_OPT_MOBILE_H_
| #define CAFFE2_OPT_MOBILE_H_
|
| #include "caffe2/core/common.h"
| #include "nomnigraph/Representations/NeuralNet.h"
|
| namespace caffe2 {
| namespace opt {
|
| CAFFE2_API void addNNPACK(nom::repr::NNModule* nn, bool low_memory = false);
| CAFFE2_API void fuseNNPACKConvRelu(nom::repr::NNModule* nn);
|
| } // namespace opt
| } // namespace caffe2
|
| #endif // CAFFE2_OPT_MOBILE_H_
|
|
