Lines Matching refs:es_index
133 for (int es_index = 0; es_index < feature_vectors.size(); ++es_index) { in ConcatEmbeddings() local
136 embedding_matrices_[es_index].get(); in ConcatEmbeddings()
139 TC_LOG(ERROR) << es_index; in ConcatEmbeddings()
144 const FeatureVector &feature_vector = feature_vectors[es_index]; in ConcatEmbeddings()
147 const int concat_offset = concat_offset_[es_index]; in ConcatEmbeddings()
151 TC_LOG(ERROR) << es_index; in ConcatEmbeddings()
159 int es_index, float *embedding) const { in GetEmbedding() argument
160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get(); in GetEmbedding()
163 TC_LOG(ERROR) << es_index; in GetEmbedding()
167 embedding_matrices_[es_index]->dim()); in GetEmbedding()
375 int EmbeddingNetwork::EmbeddingSize(int es_index) const { in EmbeddingSize()
376 return embedding_matrices_[es_index]->dim(); in EmbeddingSize()