Searched refs:EmbedTokensPerMessage (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | actions-suggestions.h | 122 bool EmbedTokensPerMessage(const std::vector<std::vector<Token>>& tokens,
|
| D | actions-suggestions_test.cc | 1379 using ActionsSuggestions::EmbedTokensPerMessage;
1448 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1469 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1492 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
1511 EXPECT_TRUE(embedder.EmbedTokensPerMessage(tokens, &embeddings, in TEST_F()
|
| D | actions-suggestions.cc | 484 bool ActionsSuggestions::EmbedTokensPerMessage( in EmbedTokensPerMessage() function in libtextclassifier3::ActionsSuggestions
665 if (!EmbedTokensPerMessage(tokens, &token_embeddings, &max_tokens)) { in SetupModelInput()
|