3 #include "caffe2/core/operator.h" 10 : alphaInv(1.0 / op->GetSingleArgument<
float>(
"alpha", 0.005f)),
11 beta(op->GetSingleArgument<
float>(
"beta", 1.0f)),
12 lambda1(op->GetSingleArgument<
float>(
"lambda1", 0.001f)),
13 lambda2(op->GetSingleArgument<
float>(
"lambda2", 0.001f)) {}
20 template <
typename T,
class Context>
23 USE_OPERATOR_CONTEXT_FUNCTIONS;
27 !HasArgument(
"alpha") || ALPHA >= InputSize(),
28 "Cannot specify alpha by both input and argument");
30 bool RunOnDevice()
override;
34 INPUT_TAGS(VAR, N_Z, GRAD, ALPHA);
35 OUTPUT_TAGS(OUTPUT_VAR, OUTPUT_N_Z);
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 ...