Searched refs:embedding_tables_ (Results 1 – 2 of 2) sorted by relevance
178 std::map<Key, ParameterPtr> embedding_tables_; variable
244 (void)embedding_tables_.emplace(key, iter->second); in InitEmbeddingTable()507 embedding_tables_.insert(std::make_pair(count, embedding_table->cast<ParameterPtr>())); in GetEmbeddingTableParamPtr()545 for (auto embedding_table : embedding_tables_) { in SyncEmbeddingTables()