Searched refs:IndexMultiSelect (Results 1 – 1 of 1) sorted by relevance
238 __global__ void IndexMultiSelect(const int num_elements, const Index* indices, in IndexMultiSelect() function483 TF_CHECK_OK(GpuLaunchKernel(IndexMultiSelect<int, float4>, config.block_count, in DoNMS()540 IndexMultiSelect<int, int>, config.block_count, config.thread_per_block, in DoNMS()