Searched refs:token_embedding_size_ (Results 1 – 3 of 3) sorted by relevance
145 int token_embedding_size_; variable
388 token_embedding_size_ = feature_processor_->GetTokenEmbeddingSize(); in ValidateAndInitialize()630 {conversation_length, max_tokens, token_embedding_size_}); in AllocateInput()
1409 token_embedding_size_ = feature_processor_->GetTokenEmbeddingSize(); in TestingMessageEmbedder()1410 EXPECT_EQ(token_embedding_size_, 1); in TestingMessageEmbedder()