1 #ifndef CAFFE2_OPERATORS_MARGIN_RANKING_CRITERION_OP_H_ 2 #define CAFFE2_OPERATORS_MARGIN_RANKING_CRITERION_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/logging.h" 6 #include "caffe2/core/operator.h" 10 template <
class Context>
15 OP_SINGLE_ARG(
float,
"margin", margin_, 1.0) {}
16 USE_OPERATOR_CONTEXT_FUNCTIONS;
18 bool RunOnDevice()
override;
24 template <
class Context>
29 OP_SINGLE_ARG(
float,
"margin", margin_, 1.0) {}
30 USE_OPERATOR_CONTEXT_FUNCTIONS;
32 bool RunOnDevice()
override;
40 #endif // CAFFE2_OPERATORS_MARGIN_RANKING_CRITERION_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 ...