Home
last modified time | relevance | path

Searched refs:kMaxSeqLen (Results 1 – 4 of 4) sorted by relevance

/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dbert_nl_classifier.cc85 std::min(static_cast<size_t>(kMaxSeqLen - 2), in Preprocess()
99 std::vector<int> input_ids(kMaxSeqLen, 0); in Preprocess()
100 std::vector<int> input_mask(kMaxSeqLen, 0); in Preprocess()
113 PopulateTensor(std::vector<int>(kMaxSeqLen, 0), segment_ids_tensor); in Preprocess()
Dbert_nl_classifier.h55 static constexpr int kMaxSeqLen = 128; variable
/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
Dbert_question_answerer.cc223 int max_context_len = kMaxSeqLen - query_tokens.size() - 3; in Preprocess()
231 segment_ids.reserve(kMaxSeqLen); in Preprocess()
260 input_ids.reserve(kMaxSeqLen); in Preprocess()
268 input_mask.reserve(kMaxSeqLen); in Preprocess()
271 int zeros_to_pad = kMaxSeqLen - input_ids.size(); in Preprocess()
Dbert_question_answerer.h85 static constexpr int kMaxSeqLen = 384; variable