diff options
author | John Mcloughlin <john.mcloughlin@arm.com> | 2024-02-07 15:00:57 +0000 |
---|---|---|
committer | john.mcloughlin <john.mcloughlin@arm.com> | 2024-02-08 23:01:29 +0000 |
commit | 33753901d4c77c958d006fb8e4a283a9a33c4426 (patch) | |
tree | 80763d2b11826a3857fe531ec9bf8d0726087fd4 /src/backends/reference/test/RefEndToEndTests.cpp | |
parent | 5bda97349eb99151a61ab787a33e9c224ca215be (diff) | |
download | armnn-33753901d4c77c958d006fb8e4a283a9a33c4426.tar.gz |
IVGCVSW-7624 GpuFsa Op: Add Softmax operator
* Added softmax operator support
* Added test cases
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: I51d530b110c4cb812f5aab31ad1ee4022d81d19e
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") { |