1 #ifndef CAFFE2_OPERATORS_LOGIT_OP_H_ 2 #define CAFFE2_OPERATORS_LOGIT_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 6 #include "caffe2/operators/elementwise_ops.h" 10 template <
class Context>
13 : eps_(op.GetSingleArgument<
float>(
"eps", 1e-6f)) {
14 CAFFE_ENFORCE_GT(eps_, 0.0);
15 CAFFE_ENFORCE_LT(eps_, 0.5);
19 bool operator()(
const int size,
const T* X,
T* Y, Context* context)
const;
24 template <
typename T,
class Context>
27 USE_OPERATOR_CONTEXT_FUNCTIONS;
28 template <
class... Args>
31 eps_(this->
template GetSingleArgument<float>(
"eps", 1e-6f)) {}
34 bool RunOnDevice()
override;
42 #endif // CAFFE2_OPERATORS_LOGIT_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...