Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cu.cc49 constexpr int kNmsBoxesPerThread = 8 * sizeof(int); variable
58 constexpr int kNmsBoxesPerThreadModuloMask = kNmsBoxesPerThread - 1;
189 kNmsBoxesPerThread * (blockIdx.y * blockDim.y + threadIdx.y); in NMSKernel()
191 j_thread_offset += kNmsBoxesPerThread * blockDim.y * gridDim.y) { in NMSKernel()
200 for (int ib = 0; ib < kNmsBoxesPerThread; ++ib) { in NMSKernel()
215 d_delete_mask[i * bit_mask_len + j_thread_offset / kNmsBoxesPerThread] = in NMSKernel()
679 (num_boxes + kNmsBoxesPerThread - 1) / kNmsBoxesPerThread; in NmsGpu()