Searched refs:NMSKernel (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 160 void NMSKernel(const Box* d_desc_sorted_boxes, const int num_boxes, in NMSKernel() function
299 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<true>, block_dim, thread_block, 0, in NmsGpu()
303 TF_CHECK_OK(GpuLaunchKernel(NMSKernel<false>, block_dim, thread_block, 0, in NmsGpu()
|