Searched refs:total_token_count (Results 1 – 3 of 3) sorted by relevance
| /external/libtextclassifier/native/actions/ |
| D | actions-suggestions.cc | 534 std::vector<float>* embeddings, int* total_token_count) const { in EmbedAndFlattenTokens()
560 *total_token_count = 0; in EmbedAndFlattenTokens()
563 ++(*total_token_count); in EmbedAndFlattenTokens()
571 ++(*total_token_count); in EmbedAndFlattenTokens()
580 ++(*total_token_count); in EmbedAndFlattenTokens()
591 for (; *total_token_count < min_num_total_tokens; ++(*total_token_count)) { in EmbedAndFlattenTokens()
601 const int total_token_count, in AllocateInput() argument
636 {1, total_token_count}); in AllocateInput()
653 int total_token_count = 0; in SetupModelInput() local
672 &total_token_count)) { in SetupModelInput()
[all …]
|
| D | actions-suggestions_test.cc | 1533 int total_token_count = 0; in TEST_F() local
1536 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1538 EXPECT_EQ(total_token_count, 5); in TEST_F()
1556 int total_token_count = 0; in TEST_F() local
1559 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1561 EXPECT_EQ(total_token_count, 7); in TEST_F()
1581 int total_token_count = 0; in TEST_F() local
1584 embedder.EmbedAndFlattenTokens(tokens, &embeddings, &total_token_count)); in TEST_F()
1586 EXPECT_EQ(total_token_count, 3); in TEST_F()
1601 int total_token_count = 0; in TEST_F() local
[all …]
|
| D | actions-suggestions.h | 135 int* total_token_count) const;
164 const int total_token_count,
|