diff options
author | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-03 10:10:14 +0100 |
---|---|---|
committer | Kshitij Sisodia <kshitij.sisodia@arm.com> | 2022-05-03 10:10:14 +0100 |
commit | dd6d07b24bbf9023ebe8e8927be8aac3291d0f58 (patch) | |
tree | 521d8d6887aedd0282f69fc7bab447bae7cf90e7 /source/application/tensorflow-lite-micro/Model.cc | |
parent | ef90497eccf48bd725a96eb79e062ebfd4e2d618 (diff) | |
download | ml-embedded-evaluation-kit-dd6d07b24bbf9023ebe8e8927be8aac3291d0f58.tar.gz |
MLECO-3178: Fix for error reporting
TensorFlow Lite Micro's error reporting fix with some minor
wiring changes.
Change-Id: Ib011ab132c82db5809b75a7c4b983a3db2a808d6
Diffstat (limited to 'source/application/tensorflow-lite-micro/Model.cc')
-rw-r--r-- | source/application/tensorflow-lite-micro/Model.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/source/application/tensorflow-lite-micro/Model.cc b/source/application/tensorflow-lite-micro/Model.cc index adcf8d7..22a1a4d 100644 --- a/source/application/tensorflow-lite-micro/Model.cc +++ b/source/application/tensorflow-lite-micro/Model.cc @@ -32,7 +32,7 @@ arm::app::Model::Model() : m_inited (false), m_type(kTfLiteNoType) { - this->m_pErrorReporter = &this->m_uErrorReporter; + this->m_pErrorReporter = tflite::GetMicroErrorReporter(); } bool arm::app::Model::Init(tflite::MicroAllocator* allocator) @@ -108,7 +108,6 @@ bool arm::app::Model::Init(tflite::MicroAllocator* allocator) TfLiteStatus allocate_status = this->m_pInterpreter->AllocateTensors(); if (allocate_status != kTfLiteOk) { - this->m_pErrorReporter->Report("[ERROR] allocateTensors() failed"); printf_err("tensor allocation failed!\n"); delete this->m_pInterpreter; return false; |