Searched refs:embedding_lookup_ops_ (Results 1 – 2 of 2) sorted by relevance
224 embedding_lookup_ops_[key] = lookup; in InitEmbeddingTable()377 if (embedding_lookup_ops_.count(key) == 0) { in DoEmbeddingLookup()383 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in DoEmbeddingLookup()433 if (embedding_lookup_ops_.count(key) == 0) { in UpdateEmbeddings()439 std::shared_ptr<PServerKernel> table_lookup_op = embedding_lookup_ops_[key]; in UpdateEmbeddings()547 if (embedding_lookup_ops_.count(key) == 0) { in SyncEmbeddingTables()551 auto lookup = embedding_lookup_ops_[key]; in SyncEmbeddingTables()
170 std::unordered_map<Key, std::shared_ptr<PServerKernel>> embedding_lookup_ops_; variable