| /external/tflite-support/tensorflow_lite_support/cc/text/tokenizers/ |
| D | tokenizer_utils.cc | 43 const tflite::metadata::ModelMetadataExtractor* metadata_extractor) { in CheckAndLoadFirstAssociatedFile() argument
52 metadata_extractor->GetAssociatedFile( in CheckAndLoadFirstAssociatedFile()
60 const tflite::metadata::ModelMetadataExtractor* metadata_extractor) { in CreateTokenizerFromProcessUnit() argument
61 if (metadata_extractor == nullptr || tokenizer_process_unit == nullptr) { in CreateTokenizerFromProcessUnit()
73 metadata_extractor)); in CreateTokenizerFromProcessUnit()
82 options->sentencePiece_model(), metadata_extractor)); in CreateTokenizerFromProcessUnit()
92 metadata_extractor)); in CreateTokenizerFromProcessUnit()
|
| D | tokenizer_utils.h | 34 const tflite::metadata::ModelMetadataExtractor* metadata_extractor);
|
| D | BUILD | 171 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| D | object_detector.cc | 126 const ModelMetadataExtractor& metadata_extractor, in GetLabelMapIfAny() argument
135 metadata_extractor.GetAssociatedFile(labels_filename)); in GetLabelMapIfAny()
142 ASSIGN_OR_RETURN(display_names_file, metadata_extractor.GetAssociatedFile( in GetLabelMapIfAny()
149 const ModelMetadataExtractor& metadata_extractor, in GetScoreThreshold() argument
153 metadata_extractor.FindFirstProcessUnit( in GetScoreThreshold()
329 const ModelMetadataExtractor* metadata_extractor = in CheckAndSetOutputs() local
330 engine_->metadata_extractor(); in CheckAndSetOutputs()
332 if (metadata_extractor->GetModelMetadata() == nullptr || in CheckAndSetOutputs()
333 metadata_extractor->GetModelMetadata()->subgraph_metadata() == nullptr) { in CheckAndSetOutputs()
340 auto output_tensors_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
[all …]
|
| D | image_segmenter.cc | 113 const ModelMetadataExtractor& metadata_extractor, in GetLabelMapIfAny() argument
122 metadata_extractor.GetAssociatedFile(labels_filename)); in GetLabelMapIfAny()
129 ASSIGN_OR_RETURN(display_names_file, metadata_extractor.GetAssociatedFile( in GetLabelMapIfAny()
258 const ModelMetadataExtractor* metadata_extractor = in CheckAndSetOutputs() local
259 engine_->metadata_extractor(); in CheckAndSetOutputs()
261 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
274 GetLabelMapIfAny(*metadata_extractor, *output_tensor_metadata->Get(0), in CheckAndSetOutputs()
|
| D | BUILD | 29 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
65 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
99 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| D | image_classifier.cc | 165 const ModelMetadataExtractor* metadata_extractor = in CheckAndSetOutputs() local
166 engine_->metadata_extractor(); in CheckAndSetOutputs()
169 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs()
192 BuildClassificationHead(*metadata_extractor, *output_tensor, in CheckAndSetOutputs()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/utils/ |
| D | image_tensor_specs.cc | 43 const ModelMetadataExtractor& metadata_extractor) { in GetInputTensorMetadataIfAny() argument
44 if (metadata_extractor.GetModelMetadata() == nullptr || in GetInputTensorMetadataIfAny()
45 metadata_extractor.GetModelMetadata()->subgraph_metadata() == nullptr) { in GetInputTensorMetadataIfAny()
48 } else if (metadata_extractor.GetInputTensorCount() != 1) { in GetInputTensorMetadataIfAny()
55 const TensorMetadata* metadata = metadata_extractor.GetInputTensorMetadata(0); in GetInputTensorMetadataIfAny()
139 const tflite::metadata::ModelMetadataExtractor& metadata_extractor) { in BuildInputImageTensorSpecs() argument
141 GetInputTensorMetadataIfAny(metadata_extractor)); in BuildInputImageTensorSpecs()
|
| D | image_tensor_specs.h | 87 const tflite::metadata::ModelMetadataExtractor& metadata_extractor);
|
| D | BUILD | 103 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/core/ |
| D | classification_head.cc | 33 const tflite::metadata::ModelMetadataExtractor& metadata_extractor, in BuildClassificationHead() argument
48 metadata_extractor.GetAssociatedFile(labels_filename)); in BuildClassificationHead()
56 ASSIGN_OR_RETURN(display_names_file, metadata_extractor.GetAssociatedFile( in BuildClassificationHead()
100 metadata_extractor.GetAssociatedFile(score_calibration_filename)); in BuildClassificationHead()
|
| D | classification_head.h | 102 const tflite::metadata::ModelMetadataExtractor& metadata_extractor,
|
| D | BUILD | 76 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| D | base_vision_task_api.h | 81 *engine_->metadata_extractor())); in CheckAndSetInputs()
|
| /external/tflite-support/tensorflow_lite_support/metadata/cc/ |
| D | BUILD | 15 name = "metadata_extractor",
16 srcs = ["metadata_extractor.cc"],
17 hdrs = ["metadata_extractor.h"],
|
| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | nl_classifier.cc | 70 const tflite::metadata::ModelMetadataExtractor* metadata_extractor) { in CheckAndLoadFirstAssociatedFile() argument
79 metadata_extractor->GetAssociatedFile( in CheckAndLoadFirstAssociatedFile()
86 const tflite::metadata::ModelMetadataExtractor* metadata_extractor) { in CreateRegexTokenizerFromProcessUnit() argument
87 if (metadata_extractor == nullptr || tokenizer_process_unit == nullptr) { in CreateRegexTokenizerFromProcessUnit()
108 metadata_extractor)); in CreateRegexTokenizerFromProcessUnit()
|
| D | BUILD | 31 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
80 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| /external/tflite-support/tensorflow_lite_support/tools/ci_build/ |
| D | build_all.sh | 24 //tensorflow_lite_support/metadata/cc:metadata_extractor \
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | BUILD | 43 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
83 "//tensorflow_lite_support/metadata/cc:metadata_extractor",
|
| D | base_task_api.h | 45 return engine_->metadata_extractor(); in GetMetadataExtractor()
|
| D | tflite_engine.h | 137 const tflite::metadata::ModelMetadataExtractor* metadata_extractor() const { in metadata_extractor() function
|