5 #include "caffe2/core/context.h" 6 #include "caffe2/core/logging.h" 7 #include "caffe2/core/operator.h" 8 #include "caffe2/utils/math.h" 12 template <
typename T,
class Context>
17 USE_OPERATOR_CONTEXT_FUNCTIONS;
18 bool RunOnDevice()
override;
21 INPUT_TAGS(PRED, REL);
22 OUTPUT_TAGS(LOSS, DPRED);
24 void ResizeInvLogITensor(
int);
25 void ComputeDiscounts(
int*,
int);
34 template <
typename T,
class Context>
38 USE_OPERATOR_CONTEXT_FUNCTIONS;
39 bool RunOnDevice()
override;
42 INPUT_TAGS(Y, DY_CACHE, DLOSS);
Tensor is the basic class in Caffe2 that stores a contiguous memory with its shape information...
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 ...