| #include "caffe2/operators/elementwise_op.h" | |
| namespace caffe2 { | |
| // See the operations supported here: | |
| // https://eigen.tuxfamily.org/dox-devel/group__QuickRefPage.html | |
| #define EIGEN_ADD(x, y) ((x) + (y)) | |
| EIGEN_FUNCTOR(Add, EIGEN_ADD, NumericTypes, SameTypeAsInput); | |
| #undef EIGEN_ADD | |
| } |