Caffe2 - C++ API
A deep learning, cross platform ML framework
ctc_greedy_decoder_op.h
1 #ifndef CAFFE2_OPERATORS_CTC_GREEDY_DECODER_OP_H_
2 #define CAFFE2_OPERATORS_CTC_GREEDY_DECODER_OP_H_
3 
4 #include "caffe2/core/context.h"
5 #include "caffe2/core/operator.h"
6 
7 namespace caffe2 {
8 
9 template <class Context>
10 class CTCGreedyDecoderOp : public Operator<Context> {
11  public:
12  USE_OPERATOR_CONTEXT_FUNCTIONS;
13  template <class... Args>
14  explicit CTCGreedyDecoderOp(Args&&... args)
15  : Operator<Context>(std::forward<Args>(args)...) {
16  merge_repeated_ =
17  this->template GetSingleArgument<bool>("merge_repeated", true);
18  }
19 
20  bool RunOnDevice() override;
21 
22  protected:
23  bool merge_repeated_;
24  INPUT_TAGS(INPUTS, SEQ_LEN);
25  OUTPUT_TAGS(OUTPUT_LEN, VALUES);
26  // Input: X, 3D tensor; L, 1D tensor. Output: Y sparse tensor
27 };
28 
29 } // namespace caffe2
30 
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 ...
Definition: blob.h:13