1 #include "caffe2/core/context.h" 2 #include "caffe2/core/operator.h" 3 #include "caffe2/utils/math.h" 7 template <
typename T,
class Context>
10 USE_OPERATOR_CONTEXT_FUNCTIONS;
13 eps_(OperatorBase::GetSingleArgument<float>(
"eps", 1e-6f)) {}
16 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 ...