| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 38 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
|
| D | topk_v2_test.cc | 38 TopKV2OpModel(int top_k, std::initializer_list<int> input_shape, in TopKV2OpModel()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | approx_topk.cc | 111 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()
|
| D | approx_topk_shape.cc | 39 int64_t input_size, int64_t rank, int64_t top_k, float recall_target, in ApproxTopKReductionOutputSize()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | approx_topk_op.cc | 122 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/lite/examples/python/ |
| D | label_image.py | 120 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 | 197 std::unique_ptr<std::vector<int32>> top_k(filter.Extract()); in Compute() local
|
| D | nn_ops_test.cc | 1446 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 | 108 std::vector<int> top_k = GetTopKIndices(probabilities, params.k()); in Run() local
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | label_image.py | 126 top_k = results.argsort()[-5:][::-1] variable
|
| /external/tensorflow/tensorflow/lite/ |
| D | simple_memory_arena_debug_dump.cc | 85 TopKLayers(size_t top_k, size_t arena_size) in TopKLayers()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 253 const bool top_k = in Step() local
|
| /external/tensorflow/tensorflow/lite/kernels/ctc/ |
| D | ctc_beam_search.h | 248 const bool top_k = in Step() local
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 509 top_k=None, argument
|
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionOutputLayer.cpp | 294 const float nms_threshold, const float eta, const int top_k, in ApplyNMSFast()
|
| /external/ComputeLibrary/examples/ |
| D | graph_ssd_mobilenet.cpp | 376 const int top_k = 100; in create_graph_float() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 1306 top_k=None, argument
1435 top_k=None, argument
|
| /external/pdfium/third_party/libopenjpeg/ |
| D | jp2.c | 1048 OPJ_INT32 k, top_k; in opj_jp2_apply_pclr() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 5739 def top_k(input, k=1, sorted=True, name=None): # pylint: disable=redefined-builtin function
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | Types.h | 1058 int top_k() const in top_k() function
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 5488 int top_k = std::max(num_boxes, keep_top_k); in ConvertCombinedNMS() local
|