Searched refs:grid_nms (Results 1 – 3 of 3) sorted by relevance
| /dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/ |
| H A D | grid_nms.cu | 29 …__global__ void grid_nms(Span<unsigned int> mask_, Span<int> count_, View<T> bboxes_, size_type nu… in __launch_bounds__()
395 void grid_nms(const Stream& stream, Span<unsigned int> workspace, TensorSpan<int> indices, TensorSp… in grid_nms() function
434 auto kernel = raw::grid_nms<T, true, BLOCK_SIZE>; in grid_nms()
439 auto kernel = raw::grid_nms<T, false, BLOCK_SIZE>; in grid_nms()
464 template void grid_nms(const Stream& stream, Span<unsigned int> workspace, TensorSpan<int> indices,…
465 template void grid_nms(const Stream& stream, Span<unsigned int> workspace, TensorSpan<int> indices,…
|
| /dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/kernels/ |
| H A D | grid_nms.hpp | 17 void grid_nms(const csl::Stream& stream, csl::Span<unsigned int> workspace, csl::TensorSpan<int> in…
|
| /dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/ |
| H A D | detection_output.hpp | 227 …kernels::grid_nms<T>(stream, workspace, indices, count, collected_bboxes, background_class_id, nor… in forward()
|