Searched defs:cache_vocab_size (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_basic.h | 37 virtual bool MallocConstantMemory(size_t cache_vocab_size) { return true; } in MallocConstantMemory()
|
| D | ps_cache_manager.cc | 25 void PsCacheManager::InsertHashTableSize(const std::string ¶m_name, size_t cache_vocab_size, si… in InsertHashTableSize()
47 size_t cache_vocab_size, size_t embedding_size) { in ReInsertHashTableSize()
808 auto cache_vocab_size = hash_info.cache_vocab_size; in HashSwapHostToDevice() local
838 auto cache_vocab_size = hash_info.cache_vocab_size; in HashSwapDeviceToHost() local
927 auto cache_vocab_size = hash_info.cache_vocab_size; in HashSwapDeviceOut() local
955 auto cache_vocab_size = hash_info.cache_vocab_size; in HashSwapDeviceIn() local
1122 size_t cache_vocab_size = item.second.cache_vocab_size; in DumpHashTables() local
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 201 def _insert_hash_table_size(name, cache_vocab_size, embedding_size, vocab_size): argument
205 def _reinsert_hash_table_size(new_name, cur_name, cache_vocab_size, embedding_size): argument
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ascend/ |
| D | ascend_ps_cache.cc | 135 bool AscendPsCache::MallocConstantMemory(size_t cache_vocab_size) { in MallocConstantMemory()
213 … size_t cache_vocab_size, size_t embedding_size, size_t swap_out_size) { in HashSwapOut()
253 … size_t cache_vocab_size, size_t embedding_size, size_t swap_in_size) { in HashSwapIn()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 145 void PSContext::InsertHashTableSize(const std::string ¶m_name, size_t cache_vocab_size, size_t … in InsertHashTableSize()
153 size_t cache_vocab_size, size_t embedding_size) const { in ReInsertHashTableSize()
|