Home
last modified time | relevance | path

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

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/
A Dctc_beam_search.h112 std::vector<int>* top_k_indices);
217 std::vector<int>* top_k_indices) { in GetTopK() argument
222 top_k_indices->clear(); in GetTopK()
224 top_k_indices->resize(K, -1); in GetTopK()
231 (*top_k_indices)[k] = (*top_k_indices)[k - 1]; in GetTopK()
235 (*top_k_indices)[k] = j; in GetTopK()
247 std::vector<int> top_k_indices; in Step() local
256 &top_k_indices); in Step()
335 const int label = top_k ? top_k_indices[ind] : ind; in Step()

Completed in 2 milliseconds