18 #ifndef CAFFE2_FB_OPERATORS_UTILITY_OPS_H_ 19 #define CAFFE2_FB_OPERATORS_UTILITY_OPS_H_ 21 #include "caffe2/core/context.h" 22 #include "caffe2/core/operator.h" 30 template <
typename TIN,
typename TOUT,
class Context>
33 USE_OPERATOR_CONTEXT_FUNCTIONS;
37 threshold_(OperatorBase::GetSingleArgument<TIN>(
"threshold", 0)),
38 low_value_(OperatorBase::GetSingleArgument<TOUT>(
"low_value", 0)),
39 high_value_(OperatorBase::GetSingleArgument<TOUT>(
"high_value", 0)) {}
41 bool RunOnDevice()
override;
53 #endif // CAFFE2_FB_OPERATORS_UTILITY_OPS_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 ...