Searched refs:box_idx (Results 1 – 3 of 3) sorted by relevance
271 CUDA_AXIS_KERNEL_LOOP(box_idx, config.virtual_thread_count.x, X) { in InitializeDataKernel()272 d_boxes_keys_iota[img_idx * image_size + box_idx] = box_idx; in InitializeDataKernel()275 if (box_idx == 0) { in InitializeDataKernel()
314 for (int box_idx = 0; box_idx < num_boxes; ++box_idx) { in DoNMSPerClass() local315 class_scores_data.push_back(scores_data[box_idx * num_classes + class_idx]); in DoNMSPerClass()319 boxes_data[(box_idx * q + class_idx) * 4 + cid]); in DoNMSPerClass()321 class_boxes_data.push_back(boxes_data[box_idx * 4 + cid]); in DoNMSPerClass()
737 auto corner_access = [&boxes](size_t box_idx, size_t corner_idx) { in AddIoUInput() argument738 return boxes[box_idx * 4 + corner_idx]; in AddIoUInput()