Home
last modified time | relevance | path

Searched refs:max_num_tokens_per_message (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/libtextclassifier/native/actions/
H A Dactions-suggestions.cc491 std::vector<float>* embeddings, int* max_num_tokens_per_message) const { in EmbedTokensPerMessage()
493 *max_num_tokens_per_message = 0; in EmbedTokensPerMessage()
496 if (num_message_tokens > *max_num_tokens_per_message) { in EmbedTokensPerMessage()
497 *max_num_tokens_per_message = num_message_tokens; in EmbedTokensPerMessage()
502 *max_num_tokens_per_message) { in EmbedTokensPerMessage()
503 *max_num_tokens_per_message = in EmbedTokensPerMessage()
506 if (model_->feature_processor_options()->max_num_tokens_per_message() > 0 && in EmbedTokensPerMessage()
507 *max_num_tokens_per_message > in EmbedTokensPerMessage()
508 model_->feature_processor_options()->max_num_tokens_per_message()) { in EmbedTokensPerMessage()
509 *max_num_tokens_per_message = in EmbedTokensPerMessage()
[all …]
H A Dactions-suggestions_test.cc1450 int max_num_tokens_per_message = 0; in TEST_F() local
1453 &max_num_tokens_per_message)); in TEST_F()
1455 EXPECT_EQ(max_num_tokens_per_message, 3); in TEST_F()
1471 int max_num_tokens_per_message = 0; in TEST_F() local
1474 &max_num_tokens_per_message)); in TEST_F()
1476 EXPECT_EQ(max_num_tokens_per_message, 5); in TEST_F()
1489 options_->max_num_tokens_per_message = 2; in TEST_F()
1494 int max_num_tokens_per_message = 0; in TEST_F() local
1497 &max_num_tokens_per_message)); in TEST_F()
1499 EXPECT_EQ(max_num_tokens_per_message, 2); in TEST_F()
[all …]
H A Dactions-suggestions.h127 int* max_num_tokens_per_message) const;
H A Dactions_model.fbs197 // The `max_num_tokens_per_message` and `min_num_tokens_per_message` are
201 max_num_tokens_per_message:int = -1;
/aosp_15_r20/out/soong/.intermediates/external/libtextclassifier/native/libtextclassifier_fbgen_actions_actions_model/gen/actions/
Dactions_model_generated.h818 int32_t max_num_tokens_per_message = -1; member
891 int32_t max_num_tokens_per_message() const { in max_num_tokens_per_message() function
985 void add_max_num_tokens_per_message(int32_t max_num_tokens_per_message) { in add_max_num_tokens_per_message()
986 …ctionsTokenFeatureProcessorOptions::VT_MAX_NUM_TOKENS_PER_MESSAGE, max_num_tokens_per_message, -1); in add_max_num_tokens_per_message()
1031 int32_t max_num_tokens_per_message = -1,
1045 builder_.add_max_num_tokens_per_message(max_num_tokens_per_message);
1075 int32_t max_num_tokens_per_message = -1,
1100 max_num_tokens_per_message,
3578 max_num_tokens_per_message(o.max_num_tokens_per_message), in ActionsTokenFeatureProcessorOptionsT()
3600 std::swap(max_num_tokens_per_message, o.max_num_tokens_per_message);
[all …]