17 #ifndef UPSAMPLE_NEAREST_OP_H_ 18 #define UPSAMPLE_NEAREST_OP_H_ 20 #include "caffe2/core/context.h" 21 #include "caffe2/core/logging.h" 22 #include "caffe2/core/operator.h" 23 #include "caffe2/utils/math.h" 27 template <
typename T,
class Context>
32 scale_(OperatorBase::GetSingleArgument<int>(
"scale", 2)) {
35 USE_OPERATOR_CONTEXT_FUNCTIONS;
37 bool RunOnDevice()
override {
39 CAFFE_NOT_IMPLEMENTED;
46 template <
typename T,
class Context>
51 scale_(OperatorBase::GetSingleArgument<int>(
"scale", 2)) {
54 USE_OPERATOR_CONTEXT_FUNCTIONS;
56 bool RunOnDevice()
override {
58 CAFFE_NOT_IMPLEMENTED;
67 #endif // UPSAMPLE_NEAREST_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 ...