Searched refs:top_n (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | BUILD | 20 ":top_n",
26 # top_n support classes imported directly from TensorFlow.
28 name = "top_n",
30 "top_n.h",
|
| D | ctc_beam_search.h | 164 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()
|
| /external/chromium-trace/catapult/common/py_utils/py_utils/ |
| D | memory_debug.py | 44 def LogHostMemoryUsage(top_n=10, level=logging.INFO): argument
61 logging.log(level, 'Memory usage of top %i processes groups', top_n)
76 top_n, pinfos_by_names.values(), key=lambda item: item['total_mem_rss'])
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 165 int top_n = output->size(); in Decode() local
173 if (scores->rows() < batch_size_ || scores->cols() < top_n) { in Decode()
199 TopPaths(top_n, &beams, &beam_log_probabilities, merge_repeated_); in Decode()
204 CHECK_EQ(top_n, beam_log_probabilities.size()); in Decode()
207 for (int i = 0; i < top_n; ++i) { in Decode()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transfer_utils.cc | 46 const int element_count, const int top_n) { in DumpTopNFloatResults() argument
50 for (int i = 0; i < top_n; ++i) { in DumpTopNFloatResults()
|
| D | graph_transfer_utils.h | 39 const int element_count, const int top_n);
|
| /external/tensorflow/tensorflow/core/ |
| D | BUILD | 2286 "lib/gtl/top_n.h",
|