| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.cpp | 67 … const unsigned int num_detected_boxes = info.max_detections() * info.max_classes_per_detection(); in validate_arguments()
181 …> &sorted_indices, const unsigned int num_output, const unsigned int max_detections, ITensor *outp… in SaveOutputs() argument
196 for(; i < max_detections; ++i) in SaveOutputs()
220 _num_max_detected_boxes = info.max_detections() * info.max_classes_per_detection(); in configure()
246 …nsorShape(info.use_regular_nms() ? info.detection_per_class() : info.max_detections()), 1, DataTyp… in configure()
257 …_indices, info.use_regular_nms() ? info.detection_per_class() : info.max_detections(), info.nms_sc… in configure()
273 …const TensorInfo _selected_indices_info = TensorInfo(TensorShape(info.max_detections()), 1, … in validate()
275 …ded_boxes_info, &_decoded_scores_info, &_selected_indices_info, info.max_detections(), info.nms_sc… in validate()
285 const unsigned int max_detections = _info.max_detections(); in run() local
352 const auto num_output = std::min<unsigned int>(max_detections, num_selected); in run()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 90 int max_detections; member
107 op_data->max_detections = m["max_detections"].AsInt32(); in Init()
166 op_data->max_detections * op_data->max_classes_per_detection; in Prepare()
426 const std::vector<float>& scores, int max_detections, in NonMaxSuppressionSingleClassHelper() argument
440 TF_LITE_ENSURE(context, (max_detections >= 0)); in NonMaxSuppressionSingleClassHelper()
463 const int output_size = std::min(num_boxes_kept, max_detections); in NonMaxSuppressionSingleClassHelper()
512 int max_detections; member
565 nms_task_param.max_detections); in ComputeNMSResult()
579 nms_task_param.max_detections); in Run()
639 std::min(op_data->detections_per_class, op_data->max_detections); in NonMaxSuppressionMultiClassRegularHelper()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CombinedNonMaxSuppression.pbtxt | 68 A [batch_size, max_detections, 4] float32 tensor
75 A [batch_size, max_detections] float32 tensor
82 A [batch_size, max_detections] float32 tensor
|
| /external/ComputeLibrary/src/graph/nodes/ |
| D | DetectionPostProcessLayerNode.cpp | 69 … const unsigned int num_detected_box = _info.max_detections() * _info.max_classes_per_detection(); in configure_output()
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDetectionPostProcessLayer.cpp | 63 …DetectionPostProcessLayerInfo info_quantized(info.max_detections(), info.max_classes_per_detection… in configure()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 398 int max_detections = 0; in SelectResultPerBatch() local
404 max_detections = std::min(result_candidate_size, total_size_per_batch); in SelectResultPerBatch()
406 max_detections = std::min(per_batch_size, result_candidate_size); in SelectResultPerBatch()
409 final_valid_detections[batch_idx] = max_detections; in SelectResultPerBatch()
411 int curr_total_size = max_detections; in SelectResultPerBatch()
|
| /external/ComputeLibrary/arm_compute/core/ |
| D | Types.h | 1114 …DetectionPostProcessLayerInfo(unsigned int max_detections, unsigned int max_classes_per_detection,…
1116 : _max_detections(max_detections), in _max_detections() argument
1128 unsigned int max_detections() const in max_detections() function
|
| /external/ComputeLibrary/examples/ |
| D | graph_ssd_mobilenet.cpp | 667 const int max_detections = 10; in create_graph_qasymm() local
681 …DetectionPostProcessLayerInfo(max_detections, max_classes_per_detection, nms_score_threshold, nms_… in create_graph_qasymm()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-composite-functions-tf.mlir | 578 …s {tf._implements = #tf_type.func<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, ma…
599 …s {tf._implements = #tf_type.func<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, ma…
602 …s {tf._implements = #tf_type.func<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, ma…
605 …s {tf._implements = #tf_type.func<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, nu…
|
| D | ops.mlir | 635 …, h_scale = 5.000000e+00 : f32, max_classes_per_detection = 1 : i64, max_detections = 20 : i64, nm…
|
| /external/ComputeLibrary/utils/ |
| D | TypePrinter.h | 2185 os << "{MaxDetections=" << detection_info.max_detections() << ","
|