Searched refs:concat_offset (Results 1 – 2 of 2) sorted by relevance
147 const int concat_offset = concat_offset_[es_index]; in ConcatEmbeddings() local149 if (!GetEmbeddingInternal(feature_vector, embedding_matrix, concat_offset, in ConcatEmbeddings()172 EmbeddingMatrix *const embedding_matrix, const int concat_offset, in GetEmbeddingInternal() argument183 concat_offset + feature_type->base() * embedding_dim; in GetEmbeddingInternal()
206 int concat_offset, float *concat,