1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
#include "caffe2/opt/optimizer.h"
#include "caffe2/opt/converter.h"
#include "caffe2/opt/mobile.h"
#include "caffe2/opt/fusion.h"
namespace caffe2 {
namespace opt {
void workspaceOptimizations(nom::repr::NNModule* nn, Workspace* ws, int level) {
switch (level) {
case 1:
opt::fuseConvBN(nn, ws);
case 0:
default:
break;
}
}
void graphOptimzations(nom::repr::NNModule* nn, int level) {
switch (level) {
case 1:
#ifdef USE_NNPACK
opt::addNNPACK(nn, false);
opt::fuseNNPACKConvRelu(nn);
#endif
case 0:
default:
break;
}
}
NetDef optimize(NetDef net, Workspace* ws, int level) {
auto nn = convertToNNModule(net);
graphOptimzations(&nn, level);
workspaceOptimizations(&nn, ws, level);
return convertToCaffe2Proto(nn, net);
}
NetDef optimize(NetDef net, int level) {
auto nn = convertToNNModule(net);
graphOptimzations(&nn, level);
return convertToCaffe2Proto(nn, net);
}
} // namespace opt
} // namespace caffe2
|
