Searched refs:num_tokens_per_message (Results 1 – 1 of 1) sorted by relevance
725 std::vector<int> num_tokens_per_message(tokens.size()); in SetupModelInput() local727 num_tokens_per_message[i] = tokens[i].size(); in SetupModelInput()730 model_->tflite_model_spec()->input_num_tokens(), num_tokens_per_message, in SetupModelInput()