3 #include "caffe2/core/context.h" 4 #include "caffe2/core/logging.h" 5 #include "caffe2/core/operator.h" 9 template <
typename T,
class Context>
16 static_cast<T
>(OperatorBase::GetSingleArgument<float>(
"alpha", 0.01));
20 USE_OPERATOR_CONTEXT_FUNCTIONS;
22 bool RunOnDevice()
override;
28 template <
typename T,
class Context>
35 static_cast<T
>(OperatorBase::GetSingleArgument<float>(
"alpha", 0.01));
39 USE_OPERATOR_CONTEXT_FUNCTIONS;
41 bool RunOnDevice()
override;
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 ...
bool HasArgument(const string &name) const
Checks if the operator has an argument of the given name.