aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp')
-rw-r--r--tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp
index 8faf9a1261..8ce7357962 100644
--- a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp
+++ b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp
@@ -20,7 +20,7 @@
#include <chrono>
#include <fstream>
#include <iostream>
-#include <stdlib.h>
+#include <cmath>
using namespace armnnTfLiteParser;
using namespace armnn;
@@ -262,7 +262,7 @@ void CheckAccuracy(std::vector<float>* toDetector0, std::vector<float>* toDetect
// Compare abs(difference) with tolerance to check for value by value equality
for (unsigned int j = 0; j < outputs[i]->size(); ++j)
{
- compare = abs(expected[j] - outputs[i]->at(j));
+ compare = std::abs(expected[j] - outputs[i]->at(j));
if (compare > 0.001f)
{
count++;
@@ -657,4 +657,4 @@ int main(int argc, char* argv[])
ARMNN_LOG(info) << "Run completed";
return 0;
-} \ No newline at end of file
+}