Searched refs:d_cub_sort_buffer (Results 1 – 1 of 1) sorted by relevance
439 Tensor d_cub_sort_buffer; in DoNMS() local442 &d_cub_sort_buffer)); in DoNMS()468 d_cub_sort_buffer.flat<int8>().data(), cub_sort_temp_storage_bytes, in DoNMS()