Searched refs:swap_out_index_addr (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
D | gpu_ps_cache.cc | 85 …e::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, size_t, in HashSwapOut() argument 89 MS_ERROR_IF_NULL(swap_out_index_addr); in HashSwapOut() 91 reinterpret_cast<int *>(swap_out_index_addr), swap_out_size, embedding_size, in HashSwapOut()
|
D | gpu_ps_cache.h | 38 …bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, size…
|
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
D | ascend_ps_cache.cc | 212 …ndPsCache::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, in HashSwapOut() argument 216 MS_ERROR_IF_NULL(swap_out_index_addr); in HashSwapOut() 242 …(void)kernel_inputs.emplace_back(std::make_shared<Address>(swap_out_index_addr, swap_out_size * si… in HashSwapOut()
|
D | ascend_ps_cache.h | 61 …bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr, size…
|
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
D | ps_cache_basic.h | 43 …rtual bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr,
|