commit e6b0e900322a92028b2d10225cd35e5346636bc1
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Jun 08 12:22:57 2023 +0100
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Thu Jun 08 12:33:01 2023 +0100
tree da077c08bf8701027b4695d5ffecb3e4f4a69e1f
parent 6b1ecf6756fe3fc130c553b2bddec8d26cd2f0cc

Fix issue with ExecuteNetwork when running with tflite executor

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Id404893c47d43dafa743f4b9524001072b426509

tests/ExecuteNetwork/TfliteExecutor.cpp

diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp
index 8412750..6455650 100644
--- a/tests/ExecuteNetwork/TfliteExecutor.cpp
+++ b/tests/ExecuteNetwork/TfliteExecutor.cpp
@@ -77,6 +77,10 @@
     }
     else
     {
+        if (builder(&m_TfLiteInterpreter) != kTfLiteOk)
+        {
+            LogAndThrow("Error loading the model into the TfLiteInterpreter.");
+        }
         std::cout << "Running on TfLite without ArmNN delegate\n";
     }