Home
last modified time | relevance | path

Searched defs:max_detections (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Ddetection_post_process_infer.c64 const int max_detections = param->max_detections_; in DetectionPostProcessInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Ddetection_post_process_fp32.c66 int NmsSingleClass(const int num_boxes, const float *decoded_boxes, const int max_detections, const… in NmsSingleClass()
/third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/
Dmindir.cc6381 …oat> &scale, float nms_iou_threshold, float nms_score_threshold, int64_t max_detections, int64_t d… in MindIR_DetectionPostProcess_CreatePrimitive()
6569 void MindIR_DetectionPostProcess_SetMaxDetections(PrimitivePtr *primitive, int64_t max_detections) { in MindIR_DetectionPostProcess_SetMaxDetections()