Searched refs:top_k_logits (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 111 std::vector<float>* top_k_logits,
216 const int K, const Vector& input, std::vector<float>* top_k_logits, in GetTopK() argument
221 top_k_logits->clear(); in GetTopK()
223 top_k_logits->resize(K, -INFINITY); in GetTopK()
227 if (logit > (*top_k_logits)[K - 1]) { in GetTopK()
229 while (k > 0 && logit > (*top_k_logits)[k - 1]) { in GetTopK()
230 (*top_k_logits)[k] = (*top_k_logits)[k - 1]; in GetTopK()
234 (*top_k_logits)[k] = logit; in GetTopK()
239 return std::max((*top_k_logits)[0], input(num_classes_ - 1)); in GetTopK()
246 std::vector<float> top_k_logits; in Step() local
[all …]
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 113 std::vector<float>* top_k_logits,
219 const int K, const Vector& input, std::vector<float>* top_k_logits, in GetTopK() argument
224 top_k_logits->clear(); in GetTopK()
226 top_k_logits->resize(K, -INFINITY); in GetTopK()
230 if (logit > (*top_k_logits)[K - 1]) { in GetTopK()
232 while (k > 0 && logit > (*top_k_logits)[k - 1]) { in GetTopK()
233 (*top_k_logits)[k] = (*top_k_logits)[k - 1]; in GetTopK()
237 (*top_k_logits)[k] = logit; in GetTopK()
242 return std::max((*top_k_logits)[0], input(num_classes_ - 1)); in GetTopK()
249 std::vector<float> top_k_logits; in Step() local
[all …]
|