Searched refs:log_probs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 130 std::vector<float>* log_probs, bool merge_repeated) const;
398 int n, std::vector<std::vector<int>>* paths, std::vector<float>* log_probs, in TopPaths() argument
401 TFLITE_DCHECK(log_probs); in TopPaths()
403 log_probs->clear(); in TopPaths()
423 log_probs->push_back(e->newp.total); in TopPaths()
|
| D | ctc_beam_search_decoder.cc | 225 std::vector<float> log_probs; in Eval() local
243 &log_probs, merge_repeated)); in Eval()
248 log_probabilities_output[b * top_paths + bp] = log_probs[bp]; in Eval()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 133 std::vector<T>* log_probs, bool merge_repeated) const;
403 int n, std::vector<std::vector<int>>* paths, std::vector<T>* log_probs, in TopPaths() argument
406 CHECK_NOTNULL(log_probs)->clear(); in TopPaths()
427 log_probs->push_back(e->newp.total); in TopPaths()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ctc_decoder_ops.cc | 329 std::vector<T> log_probs; in Compute() local
343 &log_probs, merge_repeated_)); in Compute()
348 log_prob_t(b, bp) = log_probs[bp]; in Compute()
|