Searched refs:EmbedAndFlattenTokens (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | actions-suggestions_test.cc | 1378 using ActionsSuggestions::EmbedAndFlattenTokens;
1536 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1559 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1584 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1604 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1635 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
|
| D | actions-suggestions.h | 133 bool EmbedAndFlattenTokens(const std::vector<std::vector<Token>>& tokens,
|
| D | actions-suggestions.cc | 532 bool ActionsSuggestions::EmbedAndFlattenTokens( in EmbedAndFlattenTokens() function in libtextclassifier3::ActionsSuggestions
671 if (!EmbedAndFlattenTokens(tokens, &flattened_token_embeddings, in SetupModelInput()
|