diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2020-12-07 13:33:24 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-12-08 11:56:08 +0000 |
commit | f00f6c2c6e01cac5c452b2e585b4ceea1e85a40f (patch) | |
tree | ccd58bec8be271e6a8ab531b6eff72760ef1e135 /tests/ExecuteNetwork | |
parent | f24375df4087b7d39062d8c46b190e7abea4bc9c (diff) | |
download | armnn-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/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 8 |
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."; |