Home
last modified time | relevance | path

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

/test/mlts/benchmark/jni/
Drun_tflite.cpp155 auto* output_tensor = mTfliteInterpreter->tensor(output); in saveInferenceOutput() local
157 sink.insert(sink.end(), output_tensor->data.uint8, in saveInferenceOutput()
158 output_tensor->data.uint8 + output_tensor->bytes); in saveInferenceOutput()
164 auto* output_tensor = mTfliteInterpreter->tensor(output); in getOutputError() local
165 if (output_tensor->bytes != length) { in getOutputError()
167 output_tensor->bytes, length); in getOutputError()
173 switch (output_tensor->type) { in getOutputError()
176 elements_count = output_tensor->bytes; in getOutputError()
177 for (size_t i = 0; i < output_tensor->bytes; ++i) { in getOutputError()
187 elements_count = output_tensor->bytes / sizeof(float); in getOutputError()
[all …]