Searched defs:hash_table_addr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/parameter_cache/gpu/ |
| D | gpu_cache_mem.cc | 117 bool GPUCacheMem::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_inde… in HashSwapOut()
137 bool GPUCacheMem::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_a… in HashSwapIn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/parameter_cache/gpu/ |
| D | gpu_cache_mem.cc | 118 bool GPUCacheMem::HashSwapOut(void *hash_table_addr, void *swap_out_value_addr, void *swap_out_inde… in HashSwapOut()
138 bool GPUCacheMem::HashSwapIn(void *hash_table_addr, void *swap_in_value_addr, void *swap_in_index_a… in HashSwapIn()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_sparse_embedding_operation.cc | 51 auto hash_table_addr = reinterpret_cast<float *>(hash_info.address.addr); in PushCacheFromDeviceToLocalHost() local
107 auto hash_table_addr = reinterpret_cast<float *>(hash_info.address.addr); in PullCacheFromLocalHostToDevice() local
|
| D | device_dense_embedding_operation.cc | 103 auto hash_table_addr = reinterpret_cast<float *>(hash_info.address.addr); in PushCacheFromDeviceToLocalHost() local
159 auto hash_table_addr = reinterpret_cast<float *>(hash_info.address.addr); in PullCacheFromLocalHostToDevice() local
|
| D | embedding_cache_prefetch_actor.cc | 856 float *hash_table_addr) { in InsertLocalHostCache()
870 … const int *insert_indices, const float *insert_data, float *hash_table_addr) { in InsertLocalHostCache()
907 … const float *hash_table_addr, const int *indices_addr, in LookupLocalHostCache()
1362 auto hash_table_addr = reinterpret_cast<float *>(hash_info.address.addr); in SyncDeviceEmbeddingTable() local
|