1 #ifndef CAFFE2_OPERATORS_CTC_GREEDY_DECODER_OP_H_ 2 #define CAFFE2_OPERATORS_CTC_GREEDY_DECODER_OP_H_ 4 #include "caffe2/core/context.h" 5 #include "caffe2/core/operator.h" 9 template <
class Context>
12 USE_OPERATOR_CONTEXT_FUNCTIONS;
13 template <
class... Args>
17 this->
template GetSingleArgument<bool>(
"merge_repeated",
true);
20 bool RunOnDevice()
override;
24 INPUT_TAGS(INPUTS, SEQ_LEN);
25 OUTPUT_TAGS(OUTPUT_LEN, VALUES);
31 #endif // CAFFE2_OPERATORS_CTC_GREEDY_DECODER_OP_H_
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...