Home
last modified time | relevance | path

Searched refs:getGridNMSWorkspaceSizePerBatchItem (Results 1 – 3 of 3) sorted by relevance

/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/kernels/
H A Dgrid_nms.hpp14 std::size_t getGridNMSWorkspaceSizePerBatchItem(std::size_t num_classes, std::size_t classwise_topK…
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda4dnn/primitives/
H A Ddetection_output.hpp111 …auto workspace_per_batch_item = kernels::getGridNMSWorkspaceSizePerBatchItem(num_classes, classwis… in DetectionOutputOp()
225 …auto workspace_per_batch_item = kernels::getGridNMSWorkspaceSizePerBatchItem(num_classes, classwis… in forward()
/dports/graphics/opencv/opencv-4.5.3/modules/dnn/src/cuda/
H A Dgrid_nms.cu388 std::size_t getGridNMSWorkspaceSizePerBatchItem(std::size_t num_classes, std::size_t classwise_topK) in getGridNMSWorkspaceSizePerBatchItem() function
462 std::size_t getGridNMSWorkspaceSizePerBatchItem(std::size_t num_classes, std::size_t classwise_topK…