Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc38 const TfLiteTensor* top_k; in ResizeOutput() local
191 const auto& top_k = topc.sorted_result(); in TopK() local
212 const TfLiteTensor* top_k; in Prepare() local
243 const TfLiteTensor* top_k; in Eval() local
Dtopk_v2_test.cc38 TopKV2OpModel(int top_k, std::initializer_list<int> input_shape, in TopKV2OpModel()
/external/executorch/examples/qualcomm/oss_scripts/
Dretinanet.py120 true_boxes, true_labels, det_boxes, det_labels, tp, fp, top_k, iou_thres argument
124 def collect_data(boxes, labels, top_k=-1): argument
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dapprox_topk.cc111 absl::Span<const XlaOp> init_values, int64_t top_k, in AggregateToTopKBuilder()
158 absl::Span<const XlaOp> init_values, int64_t top_k, in ApproxTopK()
254 absl::Span<const XlaOp> init_values, int64_t top_k, in ApproxTopKFallback()
Dapprox_topk_shape.cc39 int64_t input_size, int64_t rank, int64_t top_k, float recall_target, in ApproxTopKReductionOutputSize()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dapprox_topk_op.cc122 absl::Span<const xla::XlaOp> init_values, int64_t top_k, in ApproxTopKFn()
140 absl::Span<const xla::XlaOp> init_values, int64_t top_k, in ApproxTopKFn()
/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.cc197 std::unique_ptr<std::vector<int32>> top_k(filter.Extract()); in Compute() local
Dnn_ops_test.cc1446 auto top_k = ops::TopK(root, input, input_k, ops::TopK::Sorted(true)); in BM_TopK() local
/external/tensorflow/tensorflow/lite/examples/python/
Dlabel_image.py120 top_k = results.argsort()[-5:][::-1] variable
/external/mesa3d/src/gallium/frontends/teflon/tests/
Dclassification.py127 top_k = results.argsort()[-5:][::-1] variable
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dtopk_accuracy_eval_stage.cc108 std::vector<int> top_k = GetTopKIndices(probabilities, params.k()); in Run() local
/external/tensorflow/tensorflow/examples/label_image/
Dlabel_image.py126 top_k = results.argsort()[-5:][::-1] variable
/external/tensorflow/tensorflow/lite/
Dsimple_memory_arena_debug_dump.cc85 TopKLayers(size_t top_k, size_t arena_size) in TopKLayers()
/external/tensorflow/tensorflow/core/util/ctc/
Dctc_beam_search.h253 const bool top_k = in Step() local
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/
Dexplanation.proto250 int32 top_k = 4; field
/external/tensorflow/tensorflow/lite/kernels/ctc/
Dctc_beam_search.h248 const bool top_k = in Step() local
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py509 top_k=None, argument
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/
Dexplanation.proto256 int32 top_k = 4; field
/external/googleapis/google/cloud/aiplatform/v1beta1/
Dexplanation.proto264 int32 top_k = 4; field
/external/googleapis/google/cloud/aiplatform/v1/
Dexplanation.proto263 int32 top_k = 4; field
/external/ComputeLibrary/src/runtime/CPP/functions/
DCPPDetectionOutputLayer.cpp294 const float nms_threshold, const float eta, const int top_k, in ApplyNMSFast()
/external/ComputeLibrary/examples/
Dgraph_ssd_mobilenet.cpp376 const int top_k = 100; in create_graph_float() local
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py1306 top_k=None, argument
1435 top_k=None, argument
/external/pdfium/third_party/libopenjpeg/
Djp2.c1048 OPJ_INT32 k, top_k; in opj_jp2_apply_pclr() local

12