Home
last modified time | relevance | path

Searched defs:top_k (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc38 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
Dtopk_v2_test.cc38 TopKV2OpModel(int top_k, std::initializer_list<int> input_shape, in TopKV2OpModel()
/external/tensorflow/tensorflow/lite/examples/python/
Dlabel_image.py93 top_k = results.argsort()[-5:][::-1] variable
/external/tensorflow/tensorflow/core/kernels/
Din_topk_op_test.cc36 static Graph* InTopK(int num_targets, int num_classes, T top_k) { in InTopK()
Dtopk_op.cc191 std::unique_ptr<std::vector<int32>> top_k(filter.Extract()); in Compute() local
Dnn_ops_test.cc1409 auto top_k = ops::TopK(root, input, input_k, ops::TopK::Sorted(true)); in BM_TopK() local
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dtopk_accuracy_eval_stage.cc106 std::vector<int> top_k = GetTopKIndices(probabilities, params.k()); in Run() local
/external/tensorflow/tensorflow/examples/label_image/
Dlabel_image.py135 top_k = results.argsort()[-5:][::-1] variable
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_beam_search.h253 const bool top_k = in Step() local
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py240 top_k=None, argument
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search.h248 const bool top_k = in Step() local
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py1271 top_k=None, argument
1396 top_k=None, argument
/external/pdfium/third_party/libopenjpeg20/
Djp2.c1042 OPJ_INT32 k, top_k; in opj_jp2_apply_pclr() local
/external/tensorflow/tensorflow/python/ops/
Dnn_ops.py5198 def top_k(input, k=1, sorted=True, name=None): # pylint: disable=redefined-builtin function
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6019 const int top_k = std::max(num_boxes, keep_top_k); in ConvertCombinedNMS() local