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