diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-10-16 09:29:38 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-10-16 08:30:49 +0000 |
commit | 20bea0071d507772e303eb6f1c476bf1feac9be5 (patch) | |
tree | 901f087dbdbc97991217b846b9708426f6f20813 /src/backends/reference | |
parent | 8d9590e5510b8ebdc4e0b2b31ce4b653b46fc02a (diff) | |
download | armnn-20bea0071d507772e303eb6f1c476bf1feac9be5.tar.gz |
IVGCVSW-3895 Add EndToEndLayerTest for the Reference Implementation of LOG_SOFTMAX
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I500f0b819be543dc5829990d363e5d911197c9ec
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/test/RefEndToEndTests.cpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/backends/reference/test/RefEndToEndTests.cpp b/src/backends/reference/test/RefEndToEndTests.cpp index 8b768abdee..370ef6599b 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -14,6 +14,7 @@ #include <backendsCommon/test/DetectionPostProcessEndToEndTestImpl.hpp> #include <backendsCommon/test/GatherEndToEndTestImpl.hpp> #include <backendsCommon/test/InstanceNormalizationEndToEndTestImpl.hpp> +#include <backendsCommon/test/LogSoftmaxEndToEndTestImpl.hpp> #include <backendsCommon/test/PreluEndToEndTestImpl.hpp> #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp> #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp> @@ -778,6 +779,12 @@ BOOST_AUTO_TEST_CASE(RefDetectionPostProcessFastNmsUint8Test) 1.0f, 1, 0.01f, 0, 0.5f, 0); } +// LogSoftmax +BOOST_AUTO_TEST_CASE(RefLogSoftmaxEndToEndTest) +{ + LogSoftmaxEndToEndTest(defaultBackends); +} + BOOST_AUTO_TEST_CASE(RefPreluEndToEndTestFloat32) { PreluEndToEndNegativeTest<armnn::DataType::Float32>(defaultBackends); |