aboutsummaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2020-12-07 13:33:24 +0000
committerJim Flynn <jim.flynn@arm.com>2020-12-08 11:56:08 +0000
commitf00f6c2c6e01cac5c452b2e585b4ceea1e85a40f (patch)
treeccd58bec8be271e6a8ab531b6eff72760ef1e135 /tests
parentf24375df4087b7d39062d8c46b190e7abea4bc9c (diff)
downloadarmnn-f00f6c2c6e01cac5c452b2e585b4ceea1e85a40f.tar.gz
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
Diffstat (limited to 'tests')
-rw-r--r--tests/ExecuteNetwork/ExecuteNetwork.cpp8
1 files changed, 4 insertions, 4 deletions
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.";