Searched refs:d_delete_mask (Results 1 – 1 of 1) sorted by relevance
162 int* d_delete_mask) { in NMSKernel() argument209 d_delete_mask[i * bit_mask_len + j_thread_offset / kNmsBoxesPerThread] = in NMSKernel()285 int* d_delete_mask = d_nms_mask.flat<int>().data(); in NmsGpu() local301 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()305 iou_threshold, bit_mask_len, d_delete_mask)); in NmsGpu()325 device.stream(), d_delete_mask, bit_mask_len, in NmsGpu()