Searched refs:embedding_matrices_ (Results 1 – 2 of 2) sorted by relevance
126 if (feature_vectors.size() != embedding_matrices_.size()) { in ConcatEmbeddings()128 << " != " << embedding_matrices_.size(); in ConcatEmbeddings()136 embedding_matrices_[es_index].get(); in ConcatEmbeddings()160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get(); in GetEmbedding()167 embedding_matrices_[es_index]->dim()); in GetEmbedding()331 embedding_matrices_.emplace_back(new EmbeddingMatrix(matrix)); in EmbeddingNetwork()332 const int embedding_dim = embedding_matrices_.back()->dim(); in EmbeddingNetwork()338 TC_DCHECK_EQ(concat_offset_.size(), embedding_matrices_.size()); in EmbeddingNetwork()376 return embedding_matrices_[es_index]->dim(); in EmbeddingSize()
226 std::vector<std::unique_ptr<EmbeddingMatrix>> embedding_matrices_; variable