Searched refs:FindTensorByName (Results 1 – 3 of 3) sorted by relevance
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 51 using ::tflite::task::core::FindTensorByName;
69 FindTensorByName(input_tensors, input_tensor_metadatas, kIdsTensorName); in Preprocess()
71 FindTensorByName(input_tensors, input_tensor_metadatas, kMaskTensorName); in Preprocess()
72 auto* segment_ids_tensor = FindTensorByName( in Preprocess()
129 const TfLiteTensor* scores = FindTensorByName( in Postprocess()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| D | bert_question_answerer.cc | 44 using ::tflite::task::core::FindTensorByName;
168 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
173 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
178 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
295 ? FindTensorByName(output_tensors, output_tensor_metadatas, in Postprocess()
300 ? FindTensorByName(output_tensors, output_tensor_metadatas, in Postprocess()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | task_utils.h | 161 static TensorType* FindTensorByName( in FindTensorByName() function
|