aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2022-10-04 11:53:04 +0100
committerKeith Davis <keith.davis@arm.com>2022-10-04 11:53:21 +0100
commit45b82a58bccdd7a81164d2c890e565caa6669b6d (patch)
tree330ac57de1fb5f50937090f82cd2b788fb6070d4
parent71ebf5f0b94cbe5ac3abcc6e8de8ce10753eba99 (diff)
downloadarmnn-45b82a58bccdd7a81164d2c890e565caa6669b6d.tar.gz
MLCE-545 INT8 TFLite model execution abnormal
* Signed32 missing from CompareAndPrintOutput Signed-off-by: Keith Davis <keith.davis@arm.com> Change-Id: If3c93fb0d73c566ddcf439fceaa6d629029df18f
-rw-r--r--tests/ExecuteNetwork/ArmNNExecutor.cpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ArmNNExecutor.cpp b/tests/ExecuteNetwork/ArmNNExecutor.cpp
index 1e409e8d0a..b894db8049 100644
--- a/tests/ExecuteNetwork/ArmNNExecutor.cpp
+++ b/tests/ExecuteNetwork/ArmNNExecutor.cpp
@@ -695,6 +695,11 @@ void ArmNNExecutor::CompareAndPrintResult(std::vector<const void*> otherOutput)
result = ComputeRMSE<float>(outputTensor.second.GetMemoryArea(), otherOutput[index++], size);
break;
}
+ case armnn::DataType::Signed32:
+ {
+ result = ComputeRMSE<int32_t>(outputTensor.second.GetMemoryArea(), otherOutput[index++], size);
+ break;
+ }
case armnn::DataType::QSymmS16:
{
result = ComputeRMSE<int16_t>(outputTensor.second.GetMemoryArea(), otherOutput[index++], size);