Searched refs:embedded_start_token_ (Results 1 – 3 of 3) sorted by relevance
146 std::vector<float> embedded_start_token_; variable
388 !EmbedTokenId(options->start_token_id(), &embedded_start_token_) || in ValidateAndInitialize()570 embeddings->insert(embeddings->end(), embedded_start_token_.begin(), in EmbedAndFlattenTokens()571 embedded_start_token_.end()); in EmbedAndFlattenTokens()
1411 EXPECT_TRUE(EmbedTokenId(options->start_token_id(), &embedded_start_token_)); in TestingMessageEmbedder()