Home
last modified time | relevance | path

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

/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dnl_classifier.cc336 const Vector<Offset<TensorMetadata>>* output_tensor_metadatas = in Initialize() local
340 output_tensors, output_tensor_metadatas, options.output_score_tensor_name, in Initialize()
365 if (output_tensor_metadatas && in Initialize()
366 output_tensor_metadatas->size() == output_tensors.size()) { in Initialize()
367 for (int i = 0; i < output_tensor_metadatas->size(); ++i) { in Initialize()
368 const tflite::TensorMetadata* metadata = output_tensor_metadatas->Get(i); in Initialize()
384 output_tensors, output_tensor_metadatas, in Initialize()
/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
Dbert_question_answerer.cc290 auto* output_tensor_metadatas = in Postprocess() local
294 output_tensor_metadatas in Postprocess()
295 ? FindTensorByName(output_tensors, output_tensor_metadatas, in Postprocess()
299 output_tensor_metadatas in Postprocess()
300 ? FindTensorByName(output_tensors, output_tensor_metadatas, in Postprocess()