Searched refs:embedding_dim (Results 1 – 1 of 1) sorted by relevance
174 const int embedding_dim = embedding_matrix->dim(); in GetEmbeddingInternal() local183 concat_offset + feature_type->base() * embedding_dim; in GetEmbeddingInternal()188 (feature_offset + embedding_dim > concat_size)) { in GetEmbeddingInternal()189 TC_LOG(ERROR) << fi << ": " << feature_offset << " " << embedding_dim in GetEmbeddingInternal()220 for (int i = 0; i < embedding_dim; ++i, ++quant_weights, ++concat_ptr) { in GetEmbeddingInternal()226 for (int i = 0; i < embedding_dim; ++i, ++weights, ++concat_ptr) { in GetEmbeddingInternal()332 const int embedding_dim = embedding_matrices_.back()->dim(); in EmbeddingNetwork() local333 offset_sum += embedding_dim * model->GetNumFeaturesInEmbeddingSpace(i); in EmbeddingNetwork()