Home
last modified time | relevance | path

Searched refs:DoEmbeddingLookup (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Dparameter_server.h135 void DoEmbeddingLookup(Key key, const LookupIds &lookup_ids, KVMessage *res);
Dparameter_server.cc370 void ParameterServer::DoEmbeddingLookup(Key key, const LookupIds &lookup_ids, KVMessage *res) { in DoEmbeddingLookup() function in mindspore::ps::ParameterServer
800 ps_->DoEmbeddingLookup(key, keys, &res_data); in HandleEmbeddingLookup()