Home
last modified time | relevance | path

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

/external/libtextclassifier/native/actions/
Dactions-suggestions.cc382 if (!EmbedTokenId(options->padding_token_id(), &embedded_padding_token_) || in ValidateAndInitialize()
524 embeddings->insert(embeddings->end(), embedded_padding_token_.begin(), in EmbedTokensPerMessage()
525 embedded_padding_token_.end()); in EmbedTokensPerMessage()
592 embeddings->insert(embeddings->end(), embedded_padding_token_.begin(), in EmbedAndFlattenTokens()
593 embedded_padding_token_.end()); in EmbedAndFlattenTokens()
Dactions-suggestions.h142 std::vector<float> embedded_padding_token_; variable
Dactions-suggestions_test.cc1406 EmbedTokenId(options->padding_token_id(), &embedded_padding_token_)); in TestingMessageEmbedder()