1 #ifndef CAFFE2_MAP_OP_H_ 2 #define CAFFE2_MAP_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;
17 OperatorBase::GetSingleArgument<int32_t>(
"buffer_size", 1000)),
20 bool RunOnDevice()
override;
23 using BufferDataType = std::pair<float, int>;
25 std::vector<std::vector<BufferDataType>> buffers_;
31 INPUT_TAGS(PREDICTION, LABEL);
36 BufferPredictions(
const float* Xdata,
const int* labelData,
int N,
int D);
41 #endif // CAFFE2_MAP_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 ...