Home
last modified time | relevance | path

Searched refs:output_tensor_metadata (Results 1 – 9 of 9) sorted by relevance

/external/tflite-support/tensorflow_lite_support/cc/task/vision/core/
Dclassification_head.cc34 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()
Dclassification_head.h103 const tflite::TensorMetadata& output_tensor_metadata,
/external/tflite-support/tensorflow_lite_support/cc/task/vision/
Dimage_segmenter.cc261 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()
Dimage_classifier.cc169 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/
Dmetadata_extractor.cc341 ->output_tensor_metadata(); in GetOutputTensorMetadata()
352 output_tensor_metadata = GetOutputTensorMetadata(); in GetOutputTensorCount() local
353 return output_tensor_metadata == nullptr ? 0 : output_tensor_metadata->size(); in GetOutputTensorCount()
Dmetadata_version.cc239 table->output_tensor_metadata(), min_version); in UpdateMinimumVersionForTable()
/external/tflite-support/tensorflow_lite_support/codegen/
Dandroid_java_generator.cc181 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/examples/convert/
Dmetadata.md76 …[SubGraphMetadata.output_tensor_metadata](https://github.com/tensorflow/tflite-support/blob/4cd055…
/external/tflite-support/tensorflow_lite_support/metadata/
Dmetadata_schema.fbs643 output_tensor_metadata:[TensorMetadata];