Searched refs:input_tensor_metadatas (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| H A D | bert_nl_classifier.cc | 69 auto* input_tensor_metadatas = in Preprocess() local
72 FindTensorByName(input_tensors, input_tensor_metadatas, kIdsTensorName); in Preprocess()
74 FindTensorByName(input_tensors, input_tensor_metadatas, kMaskTensorName); in Preprocess()
76 input_tensors, input_tensor_metadatas, kSegmentIdsTensorName); in Preprocess()
203 auto* input_tensor_metadatas = in InitializeFromMetadata() local
206 const auto& ids_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata()
208 const auto& mask_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata()
210 const auto& segment_ids_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| H A D | bert_question_answerer.cc | 164 auto* input_tensor_metadatas = in Preprocess() local
167 input_tensor_metadatas in Preprocess()
168 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
172 input_tensor_metadatas in Preprocess()
173 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
177 input_tensor_metadatas in Preprocess()
178 ? FindTensorByName(input_tensors, input_tensor_metadatas, in Preprocess()
|