3 #ifndef HEATMAP_MAX_KEYPOINT_OP_H_ 4 #define HEATMAP_MAX_KEYPOINT_OP_H_ 6 #include "caffe2/core/context.h" 7 #include "caffe2/core/logging.h" 8 #include "caffe2/core/operator.h" 9 #include "caffe2/utils/math.h" 13 template <
typename T,
class Context>
16 template <
class... Args>
19 should_output_softmax_(this->
template GetSingleArgument<bool>(
20 "should_output_softmax",
22 USE_OPERATOR_CONTEXT_FUNCTIONS;
24 bool RunOnDevice()
override;
27 bool should_output_softmax_ =
false;
32 #endif // HEATMAP_MAX_KEYPOINT_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...