Home
last modified time | relevance | path

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

/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc83 TfLiteTensor& output_positions = in Prepare() local
108 tflite::SetTensorToDynamic(&output_positions); in Prepare()
127 TfLiteTensor& output_positions = in Eval() local
137 const int max_output_length = output_positions.dims->data[1]; in Eval()
153 /*padding_value=*/max_encoded_position, &output_positions); in Eval()
Dtext_encoder.cc172 TfLiteTensor& output_positions = in Prepare() local
198 tflite::SetTensorToDynamic(&output_positions); in Prepare()
231 TfLiteTensor& output_positions = in Eval() local
264 &output_positions); in Eval()