Searched refs:randperm_handle_duplicate_keys (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Randperm.cu | 106 randperm_handle_duplicate_keys(keys_out, shuffled_data_, bits, n, generator); in randperm_out_cuda()
122 randperm_handle_duplicate_keys(keys_out, shuffled_data_, bits, n, generator); in randperm_out_cuda()
|
| H A D | Randperm.cuh | 43 void randperm_handle_duplicate_keys(T *keys, scalar_t *data, int bits, int64_t n, std::optional<at:… in randperm_handle_duplicate_keys() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | cuda_distributions_test.cu | 177 randperm_handle_duplicate_keys(keys, values, 8, 5, gen); in TEST()
|