Searched refs:GetTopK (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| D | imagenet_topk_eval.cc | 22 std::vector<int> GetTopK(const std::vector<float>& values, int k) { in GetTopK() function
85 std::vector<int> topK = GetTopK(probabilities, k_); in ComputeEval()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 110 float GetTopK(const int K, const Vector& input,
215 float CTCBeamSearchDecoder<CTCBeamState, CTCBeamComparer>::GetTopK( in GetTopK() function
255 max_coeff = GetTopK(label_selection_size_, raw_input, &top_k_logits, in Step()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 112 float GetTopK(const int K, const Vector& input,
218 float CTCBeamSearchDecoder<CTCBeamState, CTCBeamComparer>::GetTopK( in GetTopK() function
258 max_coeff = GetTopK(label_selection_size_, raw_input, &top_k_logits, in Step()
|