1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#ifndef CAFFE2_OPT_CONCAT_ELIM_H_
#define CAFFE2_OPT_CONCAT_ELIM_H_
#include "caffe2/core/common.h"
#include "caffe2/proto/caffe2_pb.h"
#include "nomnigraph/Representations/NeuralNet.h"
namespace caffe2 {
namespace opt {
void concatElim(nom::repr::NNModule* nn);
void concatAddMulNaNClipElim(nom::repr::NNModule* nn);
void gatherFuse8BitRowwiseQuantFloatMulLengthsSumElim(nom::repr::NNModule* nn);
} // namespace opt
} // namespace caffe2
#endif // CAFFE2_OPT_CONCAT_ELIM_H_
|
