Searched refs:EmbedAndFlattenTokens (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/actions/ |
| D | actions-suggestions_test.cc | 1029 using ActionsSuggestions::EmbedAndFlattenTokens;
1198 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1224 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1252 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1274 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1310 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
|
| D | actions-suggestions.h | 141 bool EmbedAndFlattenTokens(const std::vector<std::vector<Token>> tokens,
|
| D | actions-suggestions.cc | 648 bool ActionsSuggestions::EmbedAndFlattenTokens( in EmbedAndFlattenTokens() function in libtextclassifier3::ActionsSuggestions
788 if (!EmbedAndFlattenTokens(tokens, &flattened_token_embeddings, in SetupModelInput()
|