Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
caffe2
opt
mobile.h
1
#ifndef CAFFE2_OPT_MOBILE_H_
2
#define CAFFE2_OPT_MOBILE_H_
3
4
#include "caffe2/core/common.h"
5
#include "nomnigraph/Representations/NeuralNet.h"
6
7
namespace
caffe2
{
8
namespace
opt {
9
10
CAFFE2_API
void
addNNPACK(
nom::repr::NNModule
* nn,
bool
low_memory =
false
);
11
CAFFE2_API
void
fuseNNPACKConvRelu(
nom::repr::NNModule
* nn);
12
13
}
// namespace opt
14
}
// namespace caffe2
15
16
#endif // CAFFE2_OPT_MOBILE_H_
nom::repr::NNModule
Definition:
NeuralNet.h:247
caffe2
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...
Definition:
blob.h:13
Generated on Thu Mar 21 2019 13:06:20 for Caffe2 - C++ API by
1.8.11