Searched defs:hash_table_addr (Results 1 – 3 of 3) 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()
96 bool GPUPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_ad… in HashSwapIn()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 711 …ager::LookUpHostHashTable(size_t embedding_size, size_t indices_lens, const float *hash_table_addr, in LookUpHostHashTable()
745 const float *insert_data, float *hash_table_addr) { in InsertHostHashTable()
758 … const int *insert_indices, const float *insert_data, float *hash_table_addr) { in InsertHostHashTable()
807 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in HashSwapHostToDevice() local
837 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in HashSwapDeviceToHost() local
926 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in HashSwapDeviceOut() local
954 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in HashSwapDeviceIn() local
1096 auto hash_table_addr = reinterpret_cast<float *>(hash_info.device_address.addr); in SyncDeviceEmbeddingTable() local
|
| /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()
252 bool AscendPsCache::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index… in HashSwapIn()
|