Home
last modified time | relevance | path

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

/test/mlts/benchmark/jni/
Drun_tflite.cpp212 auto* output_tensor = mTfliteInterpreter->tensor(output); in saveInferenceOutput() local
214 sink.insert(sink.end(), output_tensor->data.uint8, in saveInferenceOutput()
215 output_tensor->data.uint8 + output_tensor->bytes); in saveInferenceOutput()
221 auto* output_tensor = mTfliteInterpreter->tensor(output); in getOutputError() local
222 if (output_tensor->bytes != length) { in getOutputError()
224 output_tensor->bytes, length); in getOutputError()
230 switch (output_tensor->type) { in getOutputError()
233 elements_count = output_tensor->bytes; in getOutputError()
234 for (size_t i = 0; i < output_tensor->bytes; ++i) { in getOutputError()
244 elements_count = output_tensor->bytes / sizeof(float); in getOutputError()
[all …]