Searched refs:cub_scratch (Results 1 – 1 of 1) sorted by relevance
337 Tensor cub_scratch; in NmsGpu() local340 &cub_scratch)); in NmsGpu()346 (void*)cub_scratch.flat<int8>().data(), // temp_storage in NmsGpu()