Searched refs:EmbedTokenId (Results 1 – 3 of 3) sorted by relevance
326 if (!EmbedTokenId(options->padding_token_id(), &embedded_padding_token_) || in ValidateAndInitialize()327 !EmbedTokenId(options->start_token_id(), &embedded_start_token_) || in ValidateAndInitialize()328 !EmbedTokenId(options->end_token_id(), &embedded_end_token_)) { in ValidateAndInitialize()413 bool ActionsSuggestions::EmbedTokenId(const int32 token_id, in EmbedTokenId() function in libtextclassifier3::ActionsSuggestions
126 bool EmbedTokenId(const int32 token_id, std::vector<float>* embedding) const;
1055 EmbedTokenId(options->padding_token_id(), &embedded_padding_token_)); in TestingMessageEmbedder()1056 EXPECT_TRUE(EmbedTokenId(options->start_token_id(), &embedded_start_token_)); in TestingMessageEmbedder()1057 EXPECT_TRUE(EmbedTokenId(options->end_token_id(), &embedded_end_token_)); in TestingMessageEmbedder()