Lines Matching refs:es_index
132 // "es_index" stands for "embedding space index".
133 for (int es_index = 0; es_index < feature_vectors.size(); ++es_index) {
134 // Access is safe by es_index loop bounds and Invariant 1.
136 embedding_matrices_[es_index].get();
139 TC_LOG(ERROR) << es_index;
143 // Access is safe due to es_index loop bounds.
144 const FeatureVector &feature_vector = feature_vectors[es_index];
146 // Access is safe by es_index loop bounds, Invariant 1, and Invariant 2.
147 const int concat_offset = concat_offset_[es_index];
151 TC_LOG(ERROR) << es_index;
159 int es_index, float *embedding) const {
160 EmbeddingMatrix *const embedding_matrix = embedding_matrices_[es_index].get();
163 TC_LOG(ERROR) << es_index;
167 embedding_matrices_[es_index]->dim());
375 int EmbeddingNetwork::EmbeddingSize(int es_index) const {
376 return embedding_matrices_[es_index]->dim();
