Searched refs:embedded_start_token_ (Results 1 – 3 of 3) sorted by relevance
151 std::vector<float> embedded_start_token_; variable
327 !EmbedTokenId(options->start_token_id(), &embedded_start_token_) || in ValidateAndInitialize()681 embeddings->insert(embeddings->end(), embedded_start_token_.begin(), in EmbedAndFlattenTokens()682 embedded_start_token_.end()); in EmbedAndFlattenTokens()
1056 EXPECT_TRUE(EmbedTokenId(options->start_token_id(), &embedded_start_token_)); in TestingMessageEmbedder()