diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2024-02-07 22:58:29 +0000 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2024-02-09 09:59:02 +0000 |
| commit | ddbda6a1ed13f7bef7e0dce07a37e91b062ce98a (patch) | |
| tree | 0dca42c431988340cdc0cdc6592731fef9a068ad /src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | |
| parent | 33753901d4c77c958d006fb8e4a283a9a33c4426 (diff) | |
| download | armnn-ddbda6a1ed13f7bef7e0dce07a37e91b062ce98a.tar.gz | |
IVGCVSW-8275 GpuFsa Op: Add Activation functions available
* Currently Sigmoid and TanH Functions are implemented.
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: If9483be9201dfe47b86acc41ec7932725ac2e39e
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp')
| -rw-r--r-- | src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp index da6431f857..06b2a71dee 100644 --- a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp @@ -5,6 +5,7 @@ #include "backendsCommon/test/EndToEndTestImpl.hpp" +#include "backendsCommon/test/ActivationEndToEndTestImpl.hpp" #include "backendsCommon/test/BatchMatMulEndToEndTestImpl.hpp" #include "backendsCommon/test/Convolution2dEndToEndTestImpl.hpp" #include "backendsCommon/test/layerTests/CastTestImpl.hpp" @@ -21,6 +22,18 @@ TEST_SUITE("GpuFsaEndToEnd") std::vector<BackendId> gpuFsaDefaultBackends = {"GpuFsa"}; +// Activation +// TanH +TEST_CASE("GpuFsaTanHEndToEndTestFloat32") +{ + ActivationEndToEndTest<DataType::Float32>(gpuFsaDefaultBackends, ActivationFunction::TanH, 1.f, 0, 1.f, 1.f); +} +// Sigmoid +TEST_CASE("GpuFsaSigmoidEndToEndTestFloat32") +{ + ActivationEndToEndTest<DataType::Float32>(gpuFsaDefaultBackends, ActivationFunction::Sigmoid); +} + // BatchMatMul TEST_CASE("RefBatchMatMulEndToEndFloat32Test") { |