Searched refs:EmbedTokensPerMessage (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/actions/ |
| D | actions-suggestions_test.cc | 1030 using ActionsSuggestions::EmbedTokensPerMessage;
1097 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1121 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1147 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1168 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
|
| D | actions-suggestions.h | 130 bool EmbedTokensPerMessage(const std::vector<std::vector<Token>>& tokens,
|
| D | actions-suggestions.cc | 600 bool ActionsSuggestions::EmbedTokensPerMessage( in EmbedTokensPerMessage() function in libtextclassifier3::ActionsSuggestions
782 if (!EmbedTokensPerMessage(tokens, &token_embeddings, &max_tokens)) { in SetupModelInput()
|