IVGCVSW-2503 Fix end to end test support for Boolean output type
Change-Id: Ie94743c769fc17d976849ba4b22af3a13561ff55
diff --git a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
index 7d2b091..d17b61e 100644
--- a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
+++ b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp
@@ -4,6 +4,8 @@
//
#pragma once
+#include "TypeUtils.hpp"
+
#include <armnn/ArmNN.hpp>
#include <armnn/INetwork.hpp>
@@ -102,7 +104,14 @@
);
}
-template<typename TInput, typename TOutput>
+template<typename T>
+bool CompareBoolean(T a, T b)
+{
+ return (a == 0 && b == 0) ||(a != 0 && b != 0);
+};
+
+template<DataType ArmnnIType, DataType ArmnnOType,
+ typename TInput = ResolveType<ArmnnIType>, typename TOutput = ResolveType<ArmnnOType>>
void EndToEndLayerTestImpl(INetworkPtr network,
const std::map<int, std::vector<TInput>>& inputTensorData,
const std::map<int, std::vector<TOutput>>& expectedOutputData,
@@ -145,8 +154,18 @@
for (auto&& it : expectedOutputData)
{
std::vector<TOutput> out = outputStorage.at(it.first);
- BOOST_TEST(it.second == out);
+ if (ArmnnOType == DataType::Boolean)
+ {
+ for (unsigned int i = 0; i < out.size(); ++i)
+ {
+ BOOST_TEST(CompareBoolean<TOutput>(it.second[i], out[i]));
+ }
+ }
+ else
+ {
+ BOOST_TEST(it.second == out);
+ }
}
}
-} // anonymous namespace
\ No newline at end of file
+} // anonymous namespace