1 #include "caffe2/operators/elementwise_op.h" 7 #define EIGEN_DIV(x, y) ((x) / (y)) 8 EIGEN_FUNCTOR(Div, EIGEN_DIV, NumericTypes, SameTypeAsInput);
11 void ElementWiseDivide(
19 ConstEigenVectorArrayMap<float> dZdataVec(dZdata, n);
20 ConstEigenVectorArrayMap<float> YdataVec(Ydata, n);
21 ConstEigenVectorArrayMap<float> ZdataVec(Zdata, n);
22 EigenVectorArrayMap<float>(dXdata, n) = dZdataVec / YdataVec;
23 EigenVectorArrayMap<float>(dYdata, n) = - (dZdataVec * ZdataVec) / YdataVec;
26 REGISTER_CPU_OPERATOR(DivGradient, DivGradientOp<CPUContext>);
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...