Searched refs:embedded_start_token_ (Results 1 – 3 of 3) sorted by relevance
143 std::vector<float> embedded_start_token_; variable
383 !EmbedTokenId(options->start_token_id(), &embedded_start_token_) || in ValidateAndInitialize()565 embeddings->insert(embeddings->end(), embedded_start_token_.begin(), in EmbedAndFlattenTokens()566 embedded_start_token_.end()); in EmbedAndFlattenTokens()
1407 EXPECT_TRUE(EmbedTokenId(options->start_token_id(), &embedded_start_token_)); in TestingMessageEmbedder()