Home
last modified time | relevance | path

Searched refs:embedded_padding_token_ (Results 1 – 3 of 3) sorted by relevance

/external/libtextclassifier/actions/
Dactions-suggestions.cc326 if (!EmbedTokenId(options->padding_token_id(), &embedded_padding_token_) || in ValidateAndInitialize()
640 embeddings->insert(embeddings->end(), embedded_padding_token_.begin(), in EmbedTokensPerMessage()
641 embedded_padding_token_.end()); in EmbedTokensPerMessage()
708 embeddings->insert(embeddings->end(), embedded_padding_token_.begin(), in EmbedAndFlattenTokens()
709 embedded_padding_token_.end()); in EmbedAndFlattenTokens()
Dactions-suggestions.h150 std::vector<float> embedded_padding_token_; variable
Dactions-suggestions_test.cc1055 EmbedTokenId(options->padding_token_id(), &embedded_padding_token_)); in TestingMessageEmbedder()