Searched defs:lookup_ids (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | pserver_kernel.h | 41 …virtual void UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, const float *updat… in UpdateEmbeddings()
|
| D | embedding_look_up_proxy_kernel.cc | 79 std::vector<int> lookup_ids(size, 0); in Launch() local
|
| D | embedding_look_up_ps_kernel.cc | 90 void EmbeddingLookUpPSKernel::UpdateEmbeddings(float *embedding_table, const size_t *lookup_ids, in UpdateEmbeddings()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ |
| D | ps_cache_manager.cc | 867 std::vector<int> lookup_ids(swap_indices_size, 0); in HashSwapHostToServer() local
900 std::vector<int> lookup_ids(swap_indices_size, 0); in HashSwapServerToHost() local
959 std::vector<int> lookup_ids(swap_in_ids_size, 0); in HashSwapDeviceIn() local
989 std::vector<int> lookup_ids(swap_out_ids_size, 0); in UpdataEmbeddingTable() local
1042 std::vector<int> lookup_ids(swap_indices_lens, 0); in SyncHostEmbeddingTable() local
1088 std::vector<int> lookup_ids(swap_indices_lens, 0); in SyncDeviceEmbeddingTable() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 370 void ParameterServer::DoEmbeddingLookup(Key key, const LookupIds &lookup_ids, KVMessage *res) { in DoEmbeddingLookup()
428 void ParameterServer::UpdateEmbeddings(const Key &key, const LookupIds &lookup_ids, const Values &v… in UpdateEmbeddings()
817 const LookupIds &lookup_ids = {input.keys().begin() + 1, input.keys().end()}; in HandleUpdateEmbeddings() local
|
| D | worker.cc | 280 void Worker::DoPSEmbeddingLookup(const Key &key, const std::vector<int> &lookup_ids, std::vector<fl… in DoPSEmbeddingLookup()
367 void Worker::UpdateEmbeddingTable(const std::vector<Key> &keys, const std::vector<int> &lookup_ids, in UpdateEmbeddingTable()
892 const uint64_t *lookup_ids = send.len().data(); in UpdateEmbeddingPartitioner() local
|