1 #ifndef CAFFE2_OPERATORS_SCALE_OP_H_ 2 #define CAFFE2_OPERATORS_SCALE_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 6 #include "caffe2/utils/math.h" 10 template <
class Context>
13 USE_OPERATOR_CONTEXT_FUNCTIONS;
16 scale_(OperatorBase::GetSingleArgument<float>(
"scale", 1.0)) {}
19 bool DoRunWithType() {
23 math::Scale<T, Context>(
27 Y->template mutable_data<T>(),
32 bool RunOnDevice()
override {
42 #endif // CAFFE2_OPERATORS_SCALE_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 ...