diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-03-11 12:52:46 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-03-18 17:50:48 +0000 |
commit | bca73e1c82438f160364a113793d0a2195c665ac (patch) | |
tree | 0ebb17f7ce546af5bf1641e7a946d424f0f87b0a /src/backends/backendsCommon/test/EndToEndTestImpl.hpp | |
parent | 3184c907b2420e6c66485529f336251b2b62aecf (diff) | |
download | armnn-bca73e1c82438f160364a113793d0a2195c665ac.tar.gz |
IVGCVSW-4444 Adding Elu end to end test
* implemented activation layer end to end test
* adds support for different tolerances in layer tests
* added tests for Elu (Ref, Cl, Neon)
Signed-off-by: Jan Eilers <jan.eilers@arm.com>
Change-Id: I81e28cfb4456e815bae2fb31f5c345134ff2432f
Diffstat (limited to 'src/backends/backendsCommon/test/EndToEndTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/EndToEndTestImpl.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp index 4221f626da..a4d0d505f8 100644 --- a/src/backends/backendsCommon/test/EndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/EndToEndTestImpl.hpp @@ -107,7 +107,7 @@ inline bool ConstantUsageUint8Test(const std::vector<BackendId>& backends) // Utility template for comparing tensor elements template<DataType ArmnnType, typename T = ResolveType<ArmnnType>> -bool Compare(T a, T b) +bool Compare(T a, T b, float tolerance = 0.000001f) { if (ArmnnType == DataType::Boolean) { @@ -119,7 +119,6 @@ bool Compare(T a, T b) // NOTE: All other types can be cast to float and compared with // a certain level of tolerance - constexpr float tolerance = 0.000001f; return std::fabs(static_cast<float>(a) - static_cast<float>(b)) <= tolerance; } @@ -143,7 +142,8 @@ template<DataType ArmnnIType, DataType ArmnnOType, void EndToEndLayerTestImpl(INetworkPtr network, const std::map<int, std::vector<TInput>>& inputTensorData, const std::map<int, std::vector<TOutput>>& expectedOutputData, - std::vector<BackendId> backends) + std::vector<BackendId> backends, + float tolerance = 0.000001f) { // Create runtime in which test will run IRuntime::CreationOptions options; @@ -184,7 +184,7 @@ void EndToEndLayerTestImpl(INetworkPtr network, std::vector<TOutput> out = outputStorage.at(it.first); for (unsigned int i = 0; i < out.size(); ++i) { - BOOST_CHECK(Compare<ArmnnOType>(it.second[i], out[i]) == true); + BOOST_CHECK(Compare<ArmnnOType>(it.second[i], out[i], tolerance) == true); } } } |