Searched refs:swap_out_value_addr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.cc | 85 bool GPUPsCache::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index… in HashSwapOut() argument
88 MS_ERROR_IF_NULL(swap_out_value_addr); in HashSwapOut()
90 …SwapOut(reinterpret_cast<float *>(hash_table_addr), reinterpret_cast<float *>(swap_out_value_addr), 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 bool AscendPsCache::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_in… in HashSwapOut() argument
215 MS_ERROR_IF_NULL(swap_out_value_addr); in HashSwapOut()
238 std::make_shared<Address>(swap_out_value_addr, swap_out_size * embedding_size * sizeof(float))}; 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 …virtual bool HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_index_ad…
|