diff options
author | Keith Davis <keith.davis@arm.com> | 2022-10-04 11:53:04 +0100 |
---|---|---|
committer | Keith Davis <keith.davis@arm.com> | 2022-10-04 11:53:21 +0100 |
commit | 45b82a58bccdd7a81164d2c890e565caa6669b6d (patch) | |
tree | 330ac57de1fb5f50937090f82cd2b788fb6070d4 /tests/ExecuteNetwork | |
parent | 71ebf5f0b94cbe5ac3abcc6e8de8ce10753eba99 (diff) | |
download | armnn-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
Diffstat (limited to 'tests/ExecuteNetwork')
-rw-r--r-- | tests/ExecuteNetwork/ArmNNExecutor.cpp | 5 |
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); |