Searched refs:kNmsBlockDim (Results 1 – 1 of 1) sorted by relevance
58 constexpr int kNmsBlockDim = 16; variable159 __launch_bounds__(kNmsBlockDim* kNmsBlockDim, 4) __global__164 __shared__ Box shared_i_boxes[kNmsBlockDim]; in NMSKernel()166 __shared__ float shared_i_areas[kNmsBlockDim]; in NMSKernel()290 int num_blocks = (num_boxes + kNmsBlockDim - 1) / kNmsBlockDim; in NmsGpu()295 thread_block.x = kNmsBlockDim; in NmsGpu()296 thread_block.y = kNmsBlockDim; in NmsGpu()