1 #ifndef CAFFE_OPERATORS_MOD_OP_H_ 2 #define CAFFE_OPERATORS_MOD_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/logging.h" 6 #include "caffe2/core/operator.h" 10 template <
class Context>
13 USE_OPERATOR_CONTEXT_FUNCTIONS;
16 divisor_ = OperatorBase::GetSingleArgument<int64_t>(
"divisor", 0);
17 CAFFE_ENFORCE_NE(divisor_, 0,
"divisor must not be 0");
18 sign_follow_divisor_ =
19 OperatorBase::GetSingleArgument<bool>(
"sign_follow_divisor",
false);
22 bool RunOnDevice()
override {
34 bool sign_follow_divisor_;
39 #endif // CAFFE_OPERATORS_MOD_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 ...