diff options
Diffstat (limited to 'src/backends/reference/test/RefEndToEndTests.cpp')
-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 2c5dc37d8e..866cff8ca3 100644 --- a/src/backends/reference/test/RefEndToEndTests.cpp +++ b/src/backends/reference/test/RefEndToEndTests.cpp @@ -38,6 +38,7 @@ #include <backendsCommon/test/ResizeEndToEndTestImpl.hpp> #include <backendsCommon/test/ReverseV2EndToEndTestImpl.hpp> #include <backendsCommon/test/SliceEndToEndTestImpl.hpp> +#include <backendsCommon/test/SoftmaxEndToEndTestImpl.hpp> #include <backendsCommon/test/SpaceToDepthEndToEndTestImpl.hpp> #include <backendsCommon/test/SplitterEndToEndTestImpl.hpp> #include <backendsCommon/test/StridedSliceAsyncEndToEndTest.hpp> @@ -1242,6 +1243,12 @@ TEST_CASE("RefLogSoftmaxEndToEndTest") LogSoftmaxEndToEndTest(defaultBackends); } +// Softmax +TEST_CASE("RefSoftmaxEndToEndTestFloat32") +{ + SoftmaxEndToEnd<armnn::DataType::Float32>(defaultBackends); +} + // Prelu TEST_CASE("RefPreluEndToEndTestFloat32") { |