Home
last modified time | relevance | path

Searched defs:hashmap_size (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc149 size_t hashmap_size, size_t col_size) { in MemCopyFromHostToCache()
175 auto hashmap_size = hashmap_tensor->shape_c()[0]; in BindAndInitCacheTensor() local
212 …shMapData(void *data, const int64_t host_size, const int64_t cache_size, const size_t hashmap_size, in InitHashMapData()
258 size_t hashmap_size = static_cast<size_t>(cache_size * kEmptyRate); in InitHashMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/
Dtensor_py.cc437 size_t hashmap_size, size_t col_size) { in MemCopyFromCacheToHost()
471 auto hashmap_size = hashmap_tensor_ptr->shape_c()[0]; in FlushFromCache() local