Searched refs:token_embedding_size_ (Results 1 – 3 of 3) sorted by relevance
153 int token_embedding_size_; variable
332 token_embedding_size_ = feature_processor_->GetTokenEmbeddingSize(); in ValidateAndInitialize()746 {conversation_length, max_tokens, token_embedding_size_}); in AllocateInput()
1058 token_embedding_size_ = feature_processor_->GetTokenEmbeddingSize(); in TestingMessageEmbedder()1059 EXPECT_EQ(token_embedding_size_, 1); in TestingMessageEmbedder()