Searched refs:concat_offset_ (Results 1 – 2 of 2) sorted by relevance
147 const int concat_offset = concat_offset_[es_index]; in ConcatEmbeddings()316 concat_offset_.push_back(offset_sum); in EmbeddingNetwork()338 TC_DCHECK_EQ(concat_offset_.size(), embedding_matrices_.size()); in EmbeddingNetwork()
229 std::vector<int> concat_offset_; variable