Home
last modified time | relevance | path

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

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/
A Dctc_beam_search.h130 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()
A Dctc_beam_search_decoder.cc225 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()

Completed in 3 milliseconds