Searched defs:embedding_dim (Results 1 – 1 of 1) sorted by relevance
174 const int embedding_dim = embedding_matrix->dim(); in GetEmbeddingInternal() local332 const int embedding_dim = embedding_matrices_.back()->dim(); in EmbeddingNetwork() local