1 #ifndef CAFFE2_OPERATORS_SELU_OP_H_ 2 #define CAFFE2_OPERATORS_SELU_OP_H_ 4 #include "caffe2/core/common_omp.h" 5 #include "caffe2/core/context.h" 6 #include "caffe2/core/logging.h" 7 #include "caffe2/core/operator.h" 11 template <
typename T,
class Context>
14 USE_OPERATOR_CONTEXT_FUNCTIONS;
18 alpha_ = OperatorBase::GetSingleArgument<T>(
19 "alpha", 1.6732632423543772848170429916717f);
20 lambda_ = OperatorBase::GetSingleArgument<T>(
21 "scale", 1.0507009873554804934193349852946f);
24 CAFFE_ENFORCE_GT(lambda_, 1.0);
27 bool RunOnDevice()
override;
34 template <
typename T,
class Context>
37 USE_OPERATOR_CONTEXT_FUNCTIONS;
40 alpha_ = OperatorBase::GetSingleArgument<T>(
41 "alpha", 1.6732632423543772848170429916717f);
42 lambda_ = OperatorBase::GetSingleArgument<T>(
43 "scale", 1.0507009873554804934193349852946f);
44 CAFFE_ENFORCE_GT(lambda_, 1.0);
47 bool RunOnDevice()
override;
56 #endif // CAFFE2_OPERATORS_SELU_OP_H_ Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...