Searched refs:EmbedTokenId (Results 1 – 3 of 3) sorted by relevance
382 if (!EmbedTokenId(options->padding_token_id(), &embedded_padding_token_) || in ValidateAndInitialize()383 !EmbedTokenId(options->start_token_id(), &embedded_start_token_) || in ValidateAndInitialize()384 !EmbedTokenId(options->end_token_id(), &embedded_end_token_)) { in ValidateAndInitialize()467 bool ActionsSuggestions::EmbedTokenId(const int32 token_id, in EmbedTokenId() function in libtextclassifier3::ActionsSuggestions
118 bool EmbedTokenId(const int32 token_id, std::vector<float>* embedding) const;
1406 EmbedTokenId(options->padding_token_id(), &embedded_padding_token_)); in TestingMessageEmbedder()1407 EXPECT_TRUE(EmbedTokenId(options->start_token_id(), &embedded_start_token_)); in TestingMessageEmbedder()1408 EXPECT_TRUE(EmbedTokenId(options->end_token_id(), &embedded_end_token_)); in TestingMessageEmbedder()