Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc47 constexpr int kNmsBoxesPerThread = 8 * sizeof(int); variable
54 constexpr int kNmsBoxesPerThreadModuloMask = kNmsBoxesPerThread - 1;
183 kNmsBoxesPerThread * (blockIdx.y * blockDim.y + threadIdx.y); in NMSKernel()
185 j_thread_offset += kNmsBoxesPerThread * blockDim.y * gridDim.y) { in NMSKernel()
194 for (int ib = 0; ib < kNmsBoxesPerThread; ++ib) { in NMSKernel()
209 d_delete_mask[i * bit_mask_len + j_thread_offset / kNmsBoxesPerThread] = in NMSKernel()
264 (num_boxes + kNmsBoxesPerThread - 1) / kNmsBoxesPerThread; in NmsGpu()