Home
last modified time | relevance | path

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

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/
A DBUILD21 ":top_n",
27 # top_n support classes imported directly from TensorFlow.
29 name = "top_n",
31 "top_n.h",
A Dctc_beam_search.h164 int top_n = output->size(); in Decode() local
171 if (scores->rows() < batch_size_ || scores->cols() < top_n) { in Decode()
196 TopPaths(top_n, &beams, &beam_log_probabilities, merge_repeated_); in Decode()
201 TFLITE_DCHECK_EQ(top_n, beam_log_probabilities.size()); in Decode()
204 for (int i = 0; i < top_n; ++i) { in Decode()

Completed in 3 milliseconds