Searched refs:CTCDecoder (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_decoder.h | 36 class CTCDecoder {
43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder() function
49 virtual ~CTCDecoder() {} in ~CTCDecoder()
72 class CTCGreedyDecoder : public CTCDecoder {
75 : CTCDecoder(num_classes, batch_size, merge_repeated) {} in CTCGreedyDecoder()
77 bool Decode(const CTCDecoder::SequenceLength& seq_len, in Decode()
78 const std::vector<CTCDecoder::Input>& input, in Decode()
79 std::vector<CTCDecoder::Output>* output, in Decode()
80 CTCDecoder::ScoreOutput* scores) override { in Decode()
|
| D | ctc_beam_search.h | 42 class CTCBeamSearchDecoder : public CTCDecoder {
90 : CTCDecoder(num_classes, batch_size, merge_repeated), in CTCDecoder() function
100 bool Decode(const CTCDecoder::SequenceLength& seq_len,
101 const std::vector<CTCDecoder::Input>& input,
102 std::vector<CTCDecoder::Output>* output,
103 CTCDecoder::ScoreOutput* scores) override;
158 const CTCDecoder::SequenceLength& seq_len, in Decode()
159 const std::vector<CTCDecoder::Input>& input, in Decode()
160 std::vector<CTCDecoder::Output>* output, ScoreOutput* scores) { in Decode()
166 [this](const CTCDecoder::Output& output) -> bool { in Decode()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_decoder.h | 36 class CTCDecoder {
43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder() function
49 virtual ~CTCDecoder() {} in ~CTCDecoder()
72 class CTCGreedyDecoder : public CTCDecoder {
75 : CTCDecoder(num_classes, batch_size, merge_repeated) {} in CTCGreedyDecoder()
77 Status Decode(const CTCDecoder::SequenceLength& seq_len, in Decode()
78 const std::vector<CTCDecoder::Input>& input, in Decode()
79 std::vector<CTCDecoder::Output>* output, in Decode()
80 CTCDecoder::ScoreOutput* scores) override { in Decode()
|
| D | ctc_beam_search_test.cc | 29 using tensorflow::ctc::CTCDecoder;
137 std::vector<CTCDecoder::Output> expected_output = { in TEST()
144 std::vector<CTCDecoder::Output> expected_dict_output = { in TEST()
159 std::vector<CTCDecoder::Output> outputs(top_paths); in TEST()
160 for (CTCDecoder::Output& output : outputs) { in TEST()
172 std::vector<CTCDecoder::Output> dict_outputs(top_paths); in TEST()
173 for (CTCDecoder::Output& output : dict_outputs) { in TEST()
209 std::vector<CTCDecoder::Output> outputs(top_paths); in TEST()
210 for (CTCDecoder::Output& output : outputs) { in TEST()
277 std::vector<CTCDecoder::Output> expected_default_output = { in TEST()
[all …]
|
| D | ctc_beam_search.h | 44 class CTCBeamSearchDecoder : public CTCDecoder {
92 : CTCDecoder(num_classes, batch_size, merge_repeated), in CTCDecoder() function
102 Status Decode(const CTCDecoder::SequenceLength& seq_len,
103 const std::vector<CTCDecoder::Input>& input,
104 std::vector<CTCDecoder::Output>* output,
105 CTCDecoder::ScoreOutput* scores) override;
159 const CTCDecoder::SequenceLength& seq_len, in Decode()
160 const std::vector<CTCDecoder::Input>& input, in Decode()
161 std::vector<CTCDecoder::Output>* output, ScoreOutput* scores) { in Decode()
167 [this](const CTCDecoder::Output& output) -> bool { in Decode()
|