Searched defs:top_k (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | topk_v2.cc | 38 const TfLiteTensor* top_k; in ResizeOutput() local 190 const auto& top_k = topc.sorted_result(); in TopK() local 211 const TfLiteTensor* top_k; in Prepare() local 241 const TfLiteTensor* top_k; in Eval() local
|
D | topk_v2_test.cc | 38 TopKV2OpModel(int top_k, std::initializer_list<int> input_shape, in TopKV2OpModel()
|
/external/tensorflow/tensorflow/lite/examples/python/ |
D | label_image.py | 93 top_k = results.argsort()[-5:][::-1] variable
|
/external/tensorflow/tensorflow/core/kernels/ |
D | in_topk_op_test.cc | 36 static Graph* InTopK(int num_targets, int num_classes, T top_k) { in InTopK()
|
D | topk_op.cc | 191 std::unique_ptr<std::vector<int32>> top_k(filter.Extract()); in Compute() local
|
D | nn_ops_test.cc | 1409 auto top_k = ops::TopK(root, input, input_k, ops::TopK::Sorted(true)); in BM_TopK() local
|
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
D | topk_accuracy_eval_stage.cc | 106 std::vector<int> top_k = GetTopKIndices(probabilities, params.k()); in Run() local
|
/external/tensorflow/tensorflow/examples/label_image/ |
D | label_image.py | 135 top_k = results.argsort()[-5:][::-1] variable
|
/external/tensorflow/tensorflow/core/util/ctc/ |
D | ctc_beam_search.h | 253 const bool top_k = in Step() local
|
/external/tensorflow/tensorflow/python/keras/utils/ |
D | metrics_utils.py | 240 top_k=None, argument
|
/external/tensorflow/tensorflow/lite/experimental/kernels/ |
D | ctc_beam_search.h | 248 const bool top_k = in Step() local
|
/external/tensorflow/tensorflow/python/keras/ |
D | metrics.py | 1271 top_k=None, argument 1396 top_k=None, argument
|
/external/pdfium/third_party/libopenjpeg20/ |
D | jp2.c | 1042 OPJ_INT32 k, top_k; in opj_jp2_apply_pclr() local
|
/external/tensorflow/tensorflow/python/ops/ |
D | nn_ops.py | 5198 def top_k(input, k=1, sorted=True, name=None): # pylint: disable=redefined-builtin function
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
D | convert_nodes.cc | 6019 const int top_k = std::max(num_boxes, keep_top_k); in ConvertCombinedNMS() local
|