Searched defs:AddEmbeddingTable (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ | ||
| D | ps_cache_manager.cc | 163 void PsCacheManager::AddEmbeddingTable() const { in AddEmbeddingTable() function in mindspore::ps::PsCacheManager |
| /third_party/mindspore/mindspore/ccsrc/ps/ | ||
| D | worker.cc | 186 void Worker::AddEmbeddingTable(const Key &key, const size_t &row_count) { in AddEmbeddingTable() function in mindspore::ps::Worker |