Home
last modified time | relevance | path

Searched refs:max_output_size (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/internal/
Dnon_max_suppression_test.cc72 const int max_output_size = 4; in TEST() local
80 boxes.data(), /**num_boxes=**/ 0, scores.data(), max_output_size, in TEST()
99 const int max_output_size = kNumBoxes; in TEST() local
107 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
115 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
127 int max_output_size; in TEST() local
135 max_output_size = 100; in TEST()
137 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
145 max_output_size = 2; in TEST()
147 boxes.data(), kNumBoxes, scores.data(), max_output_size, iou_threshold, in TEST()
[all …]
/external/ComputeLibrary/tests/validation/fixtures/
DNonMaxSuppressionFixture.h50 …void setup(TensorShape input_shape, unsigned int max_output_size, float score_threshold, float nms… in setup() argument
52 ARM_COMPUTE_ERROR_ON(max_output_size == 0); in setup()
54 const TensorShape output_shape(max_output_size); in setup()
56 …_target = compute_target(input_shape, scores_shape, output_shape, max_output_size, score_thresh… in setup()
57 …_reference = compute_reference(input_shape, scores_shape, output_shape, max_output_size, score_thr… in setup()
69 … unsigned int max_output_size, float score_threshold, float nms_threshold) in compute_target() argument
78 … nms_func.configure(&bboxes, &scores, &indices, max_output_size, score_threshold, nms_threshold); in compute_target()
103 … unsigned int max_output_size, float score_threshold, float nms_threshold) in compute_reference() argument
114 …return reference::non_max_suppression(bboxes, scores, indices, max_output_size, score_threshold, n… in compute_reference()
/external/ComputeLibrary/src/runtime/CPP/functions/
DCPPNonMaximumSuppression.cpp33 const ITensor *bboxes, const ITensor *scores, ITensor *indices, unsigned int max_output_size, in configure() argument
36 … ARM_COMPUTE_LOG_PARAMS(bboxes, scores, indices, max_output_size, score_threshold, nms_threshold); in configure()
39 k->configure(bboxes, scores, indices, max_output_size, score_threshold, nms_threshold); in configure()
44 …rInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *indices, unsigned int max_output_size, in validate() argument
47 …return CPPNonMaximumSuppressionKernel::validate(bboxes, scores, indices, max_output_size, score_th… in validate()
/external/tensorflow/tensorflow/compiler/tests/
Dimage_ops_test.py892 max_output_size = 6
903 max_output_size=max_output_size,
921 max_output_size = 128
936 max_output_size=max_output_size,
948 self.assertEqual(indices_tf.size, max_output_size)
959 max_output_size = 3
974 max_output_size=max_output_size,
987 self.assertEqual(indices_tf.size, max_output_size)
1001 max_output_size = 3
1016 max_output_size=max_output_size,
[all …]
/external/ComputeLibrary/src/core/CPP/kernels/
DCPPNonMaximumSuppressionKernel.cpp38 …bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, unsigned int max_output_size, in validate_arguments() argument
49 ARM_COMPUTE_RETURN_ERROR_ON_MSG(max_output_size == 0, "Max size cannot be 0"); in validate_arguments()
63 … unsigned int max_output_size, const float score_threshold, const float iou_threshold) in configure() argument
66 …(input_bboxes->info(), input_scores->info(), output_indices->info(), max_output_size, score_thresh… in configure()
68 …auto_init_if_empty(*output_indices->info(), TensorShape(max_output_size), 1, DataType::U8, Quantiz… in configure()
75 _max_output_size = max_output_size; in configure()
86 … unsigned int max_output_size, const float score_threshold, const float iou_threshold) in validate() argument
88 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(bboxes, scores, output_indices, max_output_size, sc… in validate()
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc196 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp() argument
203 const int output_size = max_output_size.scalar<int>()(); in DoNonMaxSuppressionOp()
634 const Tensor& max_output_size = context->input(2); in Compute() local
636 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
638 max_output_size.shape().DebugString())); in Compute()
652 DoNonMaxSuppressionOp<float>(context, scores, num_boxes, max_output_size, in Compute()
673 const Tensor& max_output_size = context->input(2); in Compute() local
675 context, TensorShapeUtils::IsScalar(max_output_size.shape()), in Compute()
677 max_output_size.shape().DebugString())); in Compute()
699 DoNonMaxSuppressionOp<T>(context, scores, num_boxes, max_output_size, in Compute()
[all …]
Dnon_max_suppression_op.cu.cc316 const Tensor& scores, const int64_t max_output_size, in DoNMS() argument
405 int len_output = pad_to_max_output ? max_output_size : 0; in DoNMS()
421 &num_to_keep, context, max_output_size, flip_boxes); in DoNMS()
428 int num_outputs = std::min(num_to_keep, (int)max_output_size); // no padding! in DoNMS()
429 if (pad_to_max_output && num_outputs != max_output_size) { in DoNMS()
431 0, TensorShape({max_output_size}), &output_indices)); in DoNMS()
432 config = GetGpuLaunchConfig(max_output_size, device); in DoNMS()
479 const Tensor& max_output_size, in CheckValidInputs() argument
481 if (!TensorShapeUtils::IsScalar(max_output_size.shape())) { in CheckValidInputs()
483 max_output_size.shape().DebugString(), in CheckValidInputs()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression_test.cc35 void SetMaxOutputSize(int max_output_size) { in SetMaxOutputSize() argument
36 PopulateTensor(input_max_output_size_, {max_output_size}); in SetMaxOutputSize()
72 const int max_output_size = -1) { in NonMaxSuppressionV4OpModel() argument
78 AddConstInput(TensorType_INT32, {max_output_size}); in NonMaxSuppressionV4OpModel()
163 const int max_output_size = -1) { in NonMaxSuppressionV5OpModel() argument
169 AddConstInput(TensorType_INT32, {max_output_size}); in NonMaxSuppressionV5OpModel()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DNonMaxSuppressionV2.pbtxt12 name: "max_output_size"
35 name: "max_output_size"
71 name: "max_output_size"
DNonMaxSuppressionV3.pbtxt12 name: "max_output_size"
39 name: "max_output_size"
79 name: "max_output_size"
DNonMaxSuppressionV4.pbtxt12 name: "max_output_size"
50 name: "max_output_size"
101 name: "max_output_size"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DNonMaxSuppressionV2.pbtxt12 name: "max_output_size"
35 name: "max_output_size"
71 name: "max_output_size"
DNonMaxSuppressionV3.pbtxt12 name: "max_output_size"
39 name: "max_output_size"
79 name: "max_output_size"
DNonMaxSuppressionV4.pbtxt12 name: "max_output_size"
50 name: "max_output_size"
101 name: "max_output_size"
/external/ComputeLibrary/arm_compute/runtime/CPP/functions/
DCPPNonMaximumSuppression.h51 …ensor *bboxes, const ITensor *scores, ITensor *indices, unsigned int max_output_size, const float …
63 …rInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *indices, unsigned int max_output_size,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_NonMaxSuppressionV5.pbtxt17 name: "max_output_size"
49 `max_output_size`. Defaults to false.
56 indices from the boxes tensor, where `M <= max_output_size`.
63 scores for each selected box, where `M <= max_output_size`. Scores only differ
92 boxes, scores, max_output_size, iou_threshold, score_threshold)
Dapi_def_NonMaxSuppressionV4.pbtxt17 name: "max_output_size"
41 `max_output_size`. Defaults to false.
48 indices from the boxes tensor, where `M <= max_output_size`.
75 boxes, scores, max_output_size, iou_threshold, score_threshold)
Dapi_def_NonMaxSuppressionWithOverlaps.pbtxt18 name: "max_output_size"
42 indices from the boxes tensor, where `M <= max_output_size`.
59 overlaps, scores, max_output_size, overlap_threshold, score_threshold)
Dapi_def_NonMaxSuppression.pbtxt17 name: "max_output_size"
27 indices from the boxes tensor, where `M <= max_output_size`.
53 boxes, scores, max_output_size, iou_threshold)
Dapi_def_NonMaxSuppressionV2.pbtxt17 name: "max_output_size"
34 indices from the boxes tensor, where `M <= max_output_size`.
55 boxes, scores, max_output_size, iou_threshold)
Dapi_def_NonMaxSuppressionV3.pbtxt17 name: "max_output_size"
41 indices from the boxes tensor, where `M <= max_output_size`.
61 boxes, scores, max_output_size, iou_threshold, score_threshold)
/external/ComputeLibrary/arm_compute/core/CPP/kernels/
DCPPNonMaximumSuppressionKernel.h68 …, const ITensor *input_scores, ITensor *output_indices, unsigned int max_output_size, const float …
80 …, const ITensorInfo *input_scores, const ITensorInfo *output_indices, unsigned int max_output_size,
/external/cronet/stable/third_party/rust/chromium_crates_io/vendor/miniz_oxide-0.8.0/src/inflate/
Dmod.rs188 max_output_size: usize, in decompress_to_vec_inner()
191 let mut ret: Vec<u8> = vec![0; input.len().saturating_mul(2).min(max_output_size)]; in decompress_to_vec_inner()
218 if ret.len() >= max_output_size { in decompress_to_vec_inner()
222 let new_len = ret.len().saturating_mul(2).min(max_output_size); in decompress_to_vec_inner()
/external/cronet/tot/third_party/rust/chromium_crates_io/vendor/miniz_oxide-0.8.0/src/inflate/
Dmod.rs188 max_output_size: usize, in decompress_to_vec_inner()
191 let mut ret: Vec<u8> = vec![0; input.len().saturating_mul(2).min(max_output_size)]; in decompress_to_vec_inner()
218 if ret.len() >= max_output_size { in decompress_to_vec_inner()
222 let new_len = ret.len().saturating_mul(2).min(max_output_size); in decompress_to_vec_inner()
/external/ComputeLibrary/tests/validation/reference/
DNonMaxSuppression.cpp140 … unsigned int max_output_size, float score_threshold, float nms_threshold) in non_max_suppression() argument
143 …t output_size = std::min(static_cast<size_t>(max_output_size), num_boxes); in non_max_suppression()
159 for(unsigned int i = selected.size(); i < max_output_size; ++i) in non_max_suppression()

123