Home
last modified time | relevance | path

Searched defs:max_output_size (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dnon_max_suppression_test.cc72 const int max_output_size = 4; in TEST() local
99 const int max_output_size = kNumBoxes; in TEST() local
127 int max_output_size; in TEST() local
170 int max_output_size; in TEST() local
207 const int max_output_size = 3; in TEST() local
239 const int max_output_size = 4; in TEST() local
275 int max_output_size = 6; in TEST() local
311 int max_output_size; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cc196 int num_boxes, const Tensor& max_output_size, in DoNonMaxSuppressionOp()
634 const Tensor& max_output_size = context->input(2); in Compute() local
673 const Tensor& max_output_size = context->input(2); in Compute() local
717 const Tensor& max_output_size = context->input(2); in Compute() local
775 const Tensor& max_output_size = context->input(2); in Compute() local
844 const Tensor& max_output_size = context->input(2); in Compute() local
921 const Tensor& max_output_size = context->input(2); in Compute() local
976 const Tensor& max_output_size = context->input(2); in Compute() local
H A Dnon_max_suppression_op.cu.cc316 const Tensor& scores, const int64_t max_output_size, in DoNMS()
479 const Tensor& max_output_size, in CheckValidInputs()
535 const Tensor& max_output_size = context->input(2); in Compute() local
574 const Tensor& max_output_size = context->input(2); in Compute() local
619 const Tensor& max_output_size = context->input(2); in Compute() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DNonMaxSuppressionFixture.h50 …void setup(TensorShape input_shape, unsigned int max_output_size, float score_threshold, float nms… in setup()
69unsigned int max_output_size, float score_threshold, float nms_threshold) in compute_target()
103unsigned int max_output_size, float score_threshold, float nms_threshold) in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dnon_max_suppression_test.cc35 void SetMaxOutputSize(int max_output_size) { in SetMaxOutputSize()
72 const int max_output_size = -1) { in NonMaxSuppressionV4OpModel()
163 const int max_output_size = -1) { in NonMaxSuppressionV5OpModel()
H A Dnon_max_suppression.cc189 void ResetUnusedElementsToZeroes(const int max_output_size, in ResetUnusedElementsToZeroes()
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A DCPPNonMaximumSuppressionKernel.cpp38 …bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, unsigned int max_output_size, in validate_arguments()
63unsigned int max_output_size, const float score_threshold, const float iou_threshold) in configure()
86unsigned int max_output_size, const float score_threshold, const float iou_threshold) in validate()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dimage_ops.cc151 ShapeHandle max_output_size; in NMSShapeFn() local
174 ShapeHandle max_output_size; in SoftNMSShapeFn() local
953 ShapeHandle max_output_size; in __anon269972ec1902() local
981 ShapeHandle max_output_size; in __anon269972ec1a02() local
1076 ShapeHandle max_output_size; in __anon269972ec1d02() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CPP/functions/
H A DCPPNonMaximumSuppression.cpp33 const ITensor *bboxes, const ITensor *scores, ITensor *indices, unsigned int max_output_size, in configure()
44 …rInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *indices, unsigned int max_output_size, in validate()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dimage_ops_impl.py3755 max_output_size, argument
3808 max_output_size, argument
3898 max_output_size, argument
5342 max_output_size, argument
5437 max_output_size, argument
5675 max_output_size, argument
H A Dimage_ops_test.py5136 def nms_func(box, score, max_output_size, iou_thres): argument
5398 def func(boxes, scores, max_output_size, iou_threshold): argument
5477 def func(boxes, scores, max_output_size, iou_threshold, score_threshold): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dnms_utils.cc46 Value max_output_size = func_.getArgument(2); in RewriteFunc() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dnon_max_suppression.h88 const float* scores, const int max_output_size, in NonMaxSuppression()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DNonMaxSuppression.cpp140unsigned int max_output_size, float score_threshold, float nms_threshold) in non_max_suppression()
/aosp_15_r20/external/python/cpython2/Modules/
D_json.c200 Py_ssize_t max_output_size; in ascii_escape_unicode() local
270 Py_ssize_t max_output_size; in ascii_escape_str() local
/aosp_15_r20/external/zopfli/src/zopflipng/lodepng/
H A Dlodepng.h278 size_t max_output_size; member
H A Dlodepng.cpp1264 unsigned btype, size_t max_output_size) { in inflateHuffmanBlock()
/aosp_15_r20/external/libxaac/encoder/
H A Dixheaace_api.c1807 WORD32 max_output_size = in ixheaace_write_audio_preroll_data() local