diff options
| -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); |