Searched refs:max_classes_per_detection (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CPP/functions/ |
| H A D | CPPDetectionPostProcessLayer.cpp | 67 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(info.max_classes_per_detection() <= 0, "The number of max classes … in validate_arguments()
69 … const unsigned int num_detected_boxes = info.max_detections() * info.max_classes_per_detection(); in validate_arguments()
226 _num_max_detected_boxes = info.max_detections() * info.max_classes_per_detection(); in configure()
253 …const unsigned int num_classes_per_box = std::min(info.max_classes_per_detection(), info.num_class… in configure()
378 …unsigned int num_classes_per_box = std::min<unsigned int>(_info.max_classes_per_detection(), _info… in run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | detection_postprocess.cc | 91 int max_classes_per_detection; // Fast Non-Max-Suppression member
108 op_data->max_classes_per_detection = m["max_classes_per_detection"].AsInt32(); in Init()
166 op_data->max_detections * op_data->max_classes_per_detection; in Prepare()
777 const int max_categories_per_anchor = op_data->max_classes_per_detection; in NonMaxSuppressionMultiClassFastHelper()
|
| H A D | detection_postprocess_test.cc | 45 const TensorData& output4, int max_classes_per_detection = 1) { in BaseDetectionPostprocessOpModel() argument
57 fbb.Int("max_classes_per_detection", max_classes_per_detection); in BaseDetectionPostprocessOpModel()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | DetectionPostProcessLayerNode.cpp | 69 … const unsigned int num_detected_box = _info.max_detections() * _info.max_classes_per_detection(); in configure_output()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
| H A D | NEDetectionPostProcessLayer.cpp | 65 …PostProcessLayerInfo info_quantized(info.max_detections(), info.max_classes_per_detection(), info.… in configure()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | Types.h | 1111 …nPostProcessLayerInfo(unsigned int max_detections, unsigned int max_classes_per_detection, float n…
1114 _max_classes_per_detection(max_classes_per_detection), in _max_detections()
1130 unsigned int max_classes_per_detection() const in max_classes_per_detection() function
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_ssd_mobilenet.cpp | 669 const int max_classes_per_detection = 1; in create_graph_qasymm() local
682 …DetectionPostProcessLayerInfo(max_detections, max_classes_per_detection, nms_score_threshold, nms_… in create_graph_qasymm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| H A D | prepare-composite-functions-tf.mlir | 578 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
599 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
602 …nc<@"TFLite_Detection_PostProcess", {max_detections = 10 : i64, max_classes_per_detection = 1 : i6…
604 // expected-warning @+1 {{max_classes_per_detection attribute is not set or not an integer}}
|
| H A D | ops.mlir | 678 …r_class = 100 : i64, device = "", h_scale = 5.000000e+00 : f32, max_classes_per_detection = 1 : i6…
|
| /aosp_15_r20/external/ComputeLibrary/utils/ |
| H A D | TypePrinter.h | 2547 << "MaxClassesPerDetection=" << detection_info.max_classes_per_detection() << ","
|