aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2023-06-08 12:22:57 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2023-06-08 12:33:01 +0100
commite6b0e900322a92028b2d10225cd35e5346636bc1 (patch)
treeda077c08bf8701027b4695d5ffecb3e4f4a69e1f
parent6b1ecf6756fe3fc130c553b2bddec8d26cd2f0cc (diff)
downloadarmnn-e6b0e900322a92028b2d10225cd35e5346636bc1.tar.gz
Fix issue with ExecuteNetwork when running with tflite executor
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: Id404893c47d43dafa743f4b9524001072b426509
-rw-r--r--tests/ExecuteNetwork/TfliteExecutor.cpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/TfliteExecutor.cpp b/tests/ExecuteNetwork/TfliteExecutor.cpp
index 8412750951..6455650404 100644
--- a/tests/ExecuteNetwork/TfliteExecutor.cpp
+++ b/tests/ExecuteNetwork/TfliteExecutor.cpp
@@ -77,6 +77,10 @@ TfLiteExecutor::TfLiteExecutor(const ExecuteNetworkParams& params, armnn::IRunti
}
else
{
+ if (builder(&m_TfLiteInterpreter) != kTfLiteOk)
+ {
+ LogAndThrow("Error loading the model into the TfLiteInterpreter.");
+ }
std::cout << "Running on TfLite without ArmNN delegate\n";
}