Searched refs:LookupIds (Results 1 – 3 of 3) sorted by relevance
135 void DoEmbeddingLookup(Key key, const LookupIds &lookup_ids, KVMessage *res);136 void UpdateEmbeddings(const Key &key, const LookupIds &lookup_ids, const Values &vals);
203 using LookupIds = std::vector<Key>; variable
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()