Home
last modified time | relevance | path

Searched refs:GetOutputTensorMetadata (Results 1 – 8 of 8) sorted by relevance

/external/tflite-support/tensorflow_lite_support/metadata/cc/
Dmetadata_extractor.h98 GetOutputTensorMetadata() const;
102 const tflite::TensorMetadata* GetOutputTensorMetadata(int index) const;
Dmetadata_extractor.cc299 ModelMetadataExtractor::GetOutputTensorMetadata() const { in GetOutputTensorMetadata() function in tflite::metadata::ModelMetadataExtractor
309 const tflite::TensorMetadata* ModelMetadataExtractor::GetOutputTensorMetadata( in GetOutputTensorMetadata() function in tflite::metadata::ModelMetadataExtractor
311 return GetItemFromVector<tflite::TensorMetadata>(GetOutputTensorMetadata(), in GetOutputTensorMetadata()
317 output_tensor_metadata = GetOutputTensorMetadata(); in GetOutputTensorCount()
/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
Dbert_nl_classifier.cc130 output_tensors, GetMetadataExtractor()->GetOutputTensorMetadata(), in Postprocess()
190 GetMetadataExtractor()->GetOutputTensorMetadata(kOutputTensorIndex)) in InitializeFromMetadata()
Dnl_classifier.cc248 output_tensors, GetMetadataExtractor()->GetOutputTensorMetadata(), in Postprocess()
337 GetMetadataExtractor()->GetOutputTensorMetadata(); in Initialize()
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_segmenter.cc261 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
Dobject_detector.cc340 auto output_tensors_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
Dimage_classifier.cc169 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
Dbert_question_answerer.cc291 GetMetadataExtractor()->GetOutputTensorMetadata(); in Postprocess()