Home
last modified time | relevance | path

Searched refs:swap_out_index_addr (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/
Dgpu_ps_cache.cc85 …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()
Dgpu_ps_cache.h38 …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/
Dascend_ps_cache.cc212 …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()
Dascend_ps_cache.h61 …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/
Dps_cache_basic.h43 …rtual bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_addr,