Searched refs:output_tensor_metadata (Results 1 – 9 of 9) sorted by relevance
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/core/ |
| D | classification_head.cc | 34 const tflite::TensorMetadata& output_tensor_metadata, in BuildClassificationHead() argument
37 if (output_tensor_metadata.name() != nullptr) { in BuildClassificationHead()
38 head.name = output_tensor_metadata.name()->str(); in BuildClassificationHead()
44 output_tensor_metadata, in BuildClassificationHead()
51 output_tensor_metadata, in BuildClassificationHead()
66 output_tensor_metadata, in BuildClassificationHead()
77 output_tensor_metadata, in BuildClassificationHead()
89 output_tensor_metadata, in BuildClassificationHead()
|
| D | classification_head.h | 103 const tflite::TensorMetadata& output_tensor_metadata,
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/ |
| D | image_segmenter.cc | 261 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs() local
262 if (output_tensor_metadata != nullptr) { in CheckAndSetOutputs()
264 if (output_tensor_metadata->size() != 1) { in CheckAndSetOutputs()
269 output_tensor_metadata->size()), in CheckAndSetOutputs()
274 GetLabelMapIfAny(*metadata_extractor, *output_tensor_metadata->Get(0), in CheckAndSetOutputs()
|
| D | image_classifier.cc | 169 output_tensor_metadata = metadata_extractor->GetOutputTensorMetadata(); in CheckAndSetOutputs() local
173 if (output_tensor_metadata != nullptr) { in CheckAndSetOutputs()
174 int num_output_tensors = output_tensor_metadata->size(); in CheckAndSetOutputs()
188 output_tensor_metadata->Get(i); in CheckAndSetOutputs()
|
| /external/tflite-support/tensorflow_lite_support/metadata/cc/ |
| D | metadata_extractor.cc | 306 ->output_tensor_metadata(); in GetOutputTensorMetadata()
317 output_tensor_metadata = GetOutputTensorMetadata(); in GetOutputTensorCount() local
318 return output_tensor_metadata == nullptr ? 0 : output_tensor_metadata->size(); in GetOutputTensorCount()
|
| D | metadata_version.cc | 202 table->output_tensor_metadata(), min_version); in UpdateMinimumVersionForTable()
|
| /external/tflite-support/tensorflow_lite_support/codegen/ |
| D | android_java_generator.cc | 181 graph->input_tensor_metadata(), graph->output_tensor_metadata()); in CreateModelInfo()
199 CreateTensorInfo(graph->output_tensor_metadata()->Get(i), in CreateModelInfo()
|
| /external/tensorflow/tensorflow/lite/g3doc/convert/ |
| D | metadata.md | 47 …[SubGraphMetadata.output_tensor_metadata](https://github.com/tensorflow/tflite-support/blob/4cd055…
|
| /external/tflite-support/tensorflow_lite_support/metadata/ |
| D | metadata_schema.fbs | 599 output_tensor_metadata:[TensorMetadata];
|