Searched refs:EmbeddingMatrix (Results 1 – 2 of 2) sorted by relevance
50 class EmbeddingMatrix {52 explicit EmbeddingMatrix(const EmbeddingNetworkParams::Matrix source_matrix) in EmbeddingMatrix() function122 TC_DISALLOW_COPY_AND_ASSIGN(EmbeddingMatrix);205 EmbeddingMatrix *embedding_matrix,226 std::vector<std::unique_ptr<EmbeddingMatrix>> embedding_matrices_;
135 EmbeddingMatrix *const embedding_matrix = in ConcatEmbeddings()160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get(); in GetEmbedding()172 EmbeddingMatrix *const embedding_matrix, const int concat_offset, in GetEmbeddingInternal()331 embedding_matrices_.emplace_back(new EmbeddingMatrix(matrix)); in EmbeddingNetwork()