Caffe2 - C++ API
A deep learning, cross platform ML framework
OperatorFusion.h
1 #ifndef NOM_TRANSFORMATIONS_OPERATORFUSION_H
2 #define NOM_TRANSFORMATIONS_OPERATORFUSION_H
3 
4 #include "nomnigraph/Graph/Graph.h"
5 #include "nomnigraph/Representations/NeuralNet.h"
6 
7 namespace nom {
8 namespace transformations {
9 
10 bool fuseConvRelu(Graph<std::unique_ptr<repr::Value>, int> *);
11 
12 } // namespace transformations
13 } // namespace nom
14 
15 #endif // NOM_TRANSFORMATIONS_OPERATORFUSION_H
Definition: Caffe2.cc:16