Searched refs:swap_in_value_addr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/gpu/ |
| D | gpu_ps_cache.cc | 96 bool GPUPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_ad… in HashSwapIn() argument
99 MS_ERROR_IF_NULL(swap_in_value_addr); in HashSwapIn()
101 …shSwapIn(reinterpret_cast<float *>(hash_table_addr), reinterpret_cast<float *>(swap_in_value_addr), in HashSwapIn()
|
| D | gpu_ps_cache.h | 40 …bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr, size_t …
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 252 bool AscendPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index… in HashSwapIn() argument
255 MS_ERROR_IF_NULL(swap_in_value_addr); in HashSwapIn()
284 std::make_shared<Address>(swap_in_value_addr, swap_in_size * embedding_size * sizeof(float))); in HashSwapIn()
|
| D | ascend_ps_cache.h | 63 …bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr, size_t …
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_basic.h | 45 virtual bool HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_addr,
|