Searched defs:embedding_matrix (Results 1 – 1 of 1) sorted by relevance
135 EmbeddingMatrix *const embedding_matrix = in ConcatEmbeddings() local160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get(); in GetEmbedding() local172 EmbeddingMatrix *const embedding_matrix, const int concat_offset, in GetEmbeddingInternal()