Searched refs:es_index (Results 1 – 2 of 2) sorted by relevance
133 for (int es_index = 0; es_index < feature_vectors.size(); ++es_index) { in ConcatEmbeddings() local136 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() argument160 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()[all …]
188 bool GetEmbedding(const FeatureVector &feature_vector, int es_index,199 int EmbeddingSize(int es_index) const;