Home
last modified time | relevance | path

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

/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dbert_nl_classifier.cc87 int input_tensor_length = input_tokens_size; in Preprocess() local
90 input_tensor_length = kMaxSeqLen; in Preprocess()
93 {1, input_tensor_length}); in Preprocess()
95 {1, input_tensor_length}); in Preprocess()
97 {1, input_tensor_length}); in Preprocess()
109 std::vector<int> input_ids(input_tensor_length, 0); in Preprocess()
110 std::vector<int> input_mask(input_tensor_length, 0); in Preprocess()
123 PopulateTensor(std::vector<int>(input_tensor_length, 0), segment_ids_tensor); in Preprocess()