Home
last modified time | relevance | path

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

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/
A Dctc_beam_search.h248 const bool top_k = in Step() local
251 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()
254 if (top_k) { in Step()
335 const int label = top_k ? top_k_indices[ind] : ind; in Step()
336 const float logit = top_k ? top_k_logits[ind] : raw_input(ind); in Step()
/AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/speech_commands/
A Dlabel_wav.py68 top_k = predictions.argsort()[-num_top_predictions:][::-1]
69 for node_id in top_k:
A Dlabel_wav_dir.py76 top_k = predictions.argsort()[-num_top_predictions:][::-1]
77 for node_id in top_k:

Completed in 4 milliseconds