Home
last modified time | relevance | path

Searched refs:beam_log_probabilities (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search.h163 std::vector<float> beam_log_probabilities; in Decode() local
196 TopPaths(top_n, &beams, &beam_log_probabilities, merge_repeated_); in Decode()
201 TFLITE_DCHECK_EQ(top_n, beam_log_probabilities.size()); in Decode()
202 TFLITE_DCHECK_EQ(beams.size(), beam_log_probabilities.size()); in Decode()
207 (*scores)(b, i) = -beam_log_probabilities[i]; in Decode()
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_beam_search.h164 std::vector<float> beam_log_probabilities; in Decode() local
199 TopPaths(top_n, &beams, &beam_log_probabilities, merge_repeated_); in Decode()
204 CHECK_EQ(top_n, beam_log_probabilities.size()); in Decode()
205 CHECK_EQ(beams.size(), beam_log_probabilities.size()); in Decode()
210 (*scores)(b, i) = -beam_log_probabilities[i]; in Decode()