3 #ifndef CAFFE2_OPERATORS_FLEXIBLE_TOP_K_H_ 4 #define CAFFE2_OPERATORS_FLEXIBLE_TOP_K_H_ 6 #include "caffe2/core/logging.h" 7 #include "caffe2/core/operator.h" 8 #include "caffe2/utils/math.h" 12 template <
typename T,
class Context>
15 USE_OPERATOR_CONTEXT_FUNCTIONS;
20 bool RunOnDevice()
override;
23 template <
typename T,
class Context>
26 USE_OPERATOR_CONTEXT_FUNCTIONS;
31 bool RunOnDevice()
override;
36 #endif // CAFFE2_OPERATORS_FLEXIBLE_TOP_K_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 ...