Searched refs:swap_in_value (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | hash_impl.cu | 33 __global__ void HashSwapIn(T *hash_table, const T *swap_in_value, const int *swap_in_index, const i… in HashSwapIn() argument
38 hash_table[hash_index * hash_dim + j] = swap_in_value[i * hash_dim + j]; in HashSwapIn()
53 void DoHashSwapIn(T *hash_table, const T *swap_in_value, const int *swap_in_index, const int index_… in DoHashSwapIn() argument
55 …HashSwapIn<<<GET_BLOCKS(index_size), GET_THREADS, 0, cuda_stream>>>(hash_table, swap_in_value, swa… in DoHashSwapIn()
63 template void DoHashSwapIn<float>(float *hash_table, const float *swap_in_value, const int *swap_in…
|
| D | hash_impl.cuh | 25 void DoHashSwapIn(T *hash_table, const T *swap_in_value, const int *swap_in_index, const int index_…
|