From f00f6c2c6e01cac5c452b2e585b4ceea1e85a40f Mon Sep 17 00:00:00 2001
From: Matthew Sloyan <matthew.sloyan@arm.com>
Date: Mon, 7 Dec 2020 13:33:24 +0000
Subject: IVGCVSW-5560 Fix TfLiteDelegate Reshape operator failure

 * Fixed issue when running certain models with 2D shape tensor.
 * Falls back to inbuilt options if encountered.
 * Fixed ExecuteNetwork so that error messages are logged if NULL.
 * Updated TfLiteDelegate docs to include Logical Operators.

Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Signed-off-by: David Monahan <david.monahan@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I5dbaf30351f7fc86e6178a0caf46c152812088d3
---
 tests/ExecuteNetwork/ExecuteNetwork.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'tests/ExecuteNetwork')

diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 00507e0c49..6d60eaf40d 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -89,7 +89,7 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params,
         {
             auto inputData = tfLiteInterpreter->typed_tensor<float>(input);
 
-            if(tfLiteInterpreter == NULL)
+            if(inputData == NULL)
             {
                 ARMNN_LOG(fatal) << "Input tensor is null, input type: "
                                     "\"" << params.m_InputTypes[inputIndex] << "\" may be incorrect.";
@@ -109,7 +109,7 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params,
         {
             auto inputData = tfLiteInterpreter->typed_tensor<int8_t>(input);
 
-            if(tfLiteInterpreter == NULL)
+            if(inputData == NULL)
             {
                 ARMNN_LOG(fatal) << "Input tensor is null, input type: "
                                     "\"" << params.m_InputTypes[inputIndex] << "\" may be incorrect.";
@@ -129,7 +129,7 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params,
         {
             auto inputData = tfLiteInterpreter->typed_tensor<int32_t>(input);
 
-            if(tfLiteInterpreter == NULL)
+            if(inputData == NULL)
             {
                 ARMNN_LOG(fatal) << "Input tensor is null, input type: "
                                     "\"" << params.m_InputTypes[inputIndex] << "\" may be incorrect.";
@@ -149,7 +149,7 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params,
         {
             auto inputData = tfLiteInterpreter->typed_tensor<uint8_t>(input);
 
-            if(tfLiteInterpreter == NULL)
+            if(inputData == NULL)
             {
                 ARMNN_LOG(fatal) << "Input tensor is null, input type: "
                                     "\"" << params.m_InputTypes[inputIndex] << "\" may be incorrect.";
-- 
cgit v1.2.1