Mojo function

non_max_suppression_shape_func

non_max_suppression_shape_func[type: DType](boxes: NDBuffer[type, 3, origin], scores: NDBuffer[type, 3, origin], max_output_boxes_per_class: Int, iou_threshold: SIMD[float32, 1], score_threshold: SIMD[float32, 1]) -> Index[2]

Overload to compute the output shape. Can be removed once the graph compiler supports value semantic kernels that allocate their own output.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit