Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cu.cc117 __global__ void NMSReduce(const int* bitmask, const int bit_mask_len, in NMSReduce() function
324 TF_CHECK_OK(GpuLaunchKernel(NMSReduce, 1, 1024, bit_mask_len * sizeof(int), in NmsGpu()