Home
last modified time | relevance | path

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

/external/libtextclassifier/native/utils/tflite/
Dtoken_encoder.cc135 std::vector<int> encoded_positions; in Eval() local
147 encoded_positions.push_back(std::min(k, max_encoded_position - 1)); in Eval()
152 max_output_length, encoded_positions, in Eval()
156 output_lengths.data.i32[0] = encoded_positions.size() - num_skip; in Eval()
Dtext_encoder.cc237 std::vector<int> encoded_positions; in Eval() local
253 encoded_positions.push_back(std::min(i, max_encoded_position - 1)); in Eval()
263 CopyDataToTensorAndPadOrTruncate(max_output_length, encoded_positions, in Eval()