1 #ifndef CAFFE2_OPERATORS_ELEMENTWISE_DIV_OP_H_ 2 #define CAFFE2_OPERATORS_ELEMENTWISE_DIV_OP_H_ 6 #include "caffe2/operators/elementwise_ops.h" 7 #include "caffe2/utils/math.h" 11 template <
class Context>
13 template <
typename TIn,
typename TOut>
15 const std::vector<int>& A_dims,
16 const std::vector<int>& B_dims,
20 Context* context)
const {
33 template <
typename TGrad,
typename TIn,
typename TOut>
35 const std::vector<int>& A_dims,
36 const std::vector<int>& B_dims,
43 Context* context)
const;
48 #endif // CAFFE2_OPERATORS_ELEMENTWISE_DIV_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...