1 #ifndef CAFFE2_OPERATORS_ACCURACY_OP_H_ 2 #define CAFFE2_OPERATORS_ACCURACY_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 9 template <
typename T,
class Context>
12 USE_OPERATOR_CONTEXT_FUNCTIONS;
15 top_k_(OperatorBase::GetSingleArgument<int>(
"top_k", 1)) {}
17 bool RunOnDevice()
override;
21 INPUT_TAGS(PREDICTION, LABEL);
26 #endif // CAFFE2_OPERATORS_ACCURACY_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 ...