Searched refs:randperm_handle_duplicate_keys_kernel (Results 1 – 1 of 1) sorted by relevance
13 __global__ void randperm_handle_duplicate_keys_kernel(T *keys, scalar_t *data, T mask, int n, at::P… in randperm_handle_duplicate_keys_kernel() function53 …randperm_handle_duplicate_keys_kernel<<<(n + 511) / 512, 512, 0, at::cuda::getCurrentCUDAStream()>… in randperm_handle_duplicate_keys()