Searched refs:bit_mask_len (Results 1 – 1 of 1) sorted by relevance
117 __global__ void NMSReduce(const int* bitmask, const int bit_mask_len, in NMSReduce() argument122 for (int box : GpuGridRangeX(bit_mask_len)) { in NMSReduce()133 int offset = box * bit_mask_len; in NMSReduce()135 for (int b : GpuGridRangeX(bit_mask_len)) { in NMSReduce()161 const float iou_threshold, const int bit_mask_len, in NMSKernel() argument209 d_delete_mask[i * bit_mask_len + j_thread_offset / kNmsBoxesPerThread] = in NMSKernel()263 const int bit_mask_len = in NmsGpu() local266 int64 max_nms_mask_size = num_boxes * bit_mask_len; in NmsGpu()301 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()305 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()[all …]