3 #ifndef CAFFE2_OPERATORS_BATCH_SPARSE_TO_DENSE_OP_H_ 4 #define CAFFE2_OPERATORS_BATCH_SPARSE_TO_DENSE_OP_H_ 6 #include "caffe2/core/context.h" 7 #include "caffe2/core/operator.h" 8 #include "caffe2/utils/math.h" 12 template <
typename T,
class Context>
15 USE_OPERATOR_CONTEXT_FUNCTIONS;
18 OP_SINGLE_ARG(TIndex,
"dense_last_dim", dense_last_dim_, -1),
19 OP_SINGLE_ARG(T,
"default_value", default_value_, static_cast<T>(0)) {}
20 bool RunOnDevice()
override;
23 TIndex dense_last_dim_;
25 INPUT_TAGS(LENGTHS, INDICES, VALUES);
28 template <
typename T,
class Context>
31 USE_OPERATOR_CONTEXT_FUNCTIONS;
34 bool RunOnDevice()
override;
37 TIndex dense_last_dim_;
38 INPUT_TAGS(LENGTHS, INDICES, DENSE);
43 #endif // CAFFE2_OPERATORS_BATCH_SPARSE_TO_DENSE_OP_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 ...