From 03fbeaf532f2575381edc2336f834973117f6e0f Mon Sep 17 00:00:00 2001 From: Colm Donelan Date: Wed, 26 Feb 2020 15:39:23 +0000 Subject: IVGCVSW-4440 : Add HARD_SWISH Support to Activation in CpuRef * Add a new Activiation type of HardSwish. * Add CpuRef support and tests. Signed-off-by: Colm Donelan Change-Id: I68c3840aa45b7a27d5e416a5d50fe8f99f003ce8 --- .../test/layerTests/ActivationTestImpl.hpp | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'src/backends/backendsCommon/test/layerTests/ActivationTestImpl.hpp') diff --git a/src/backends/backendsCommon/test/layerTests/ActivationTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/ActivationTestImpl.hpp index 28301188d5..2bd517180f 100644 --- a/src/backends/backendsCommon/test/layerTests/ActivationTestImpl.hpp +++ b/src/backends/backendsCommon/test/layerTests/ActivationTestImpl.hpp @@ -216,6 +216,22 @@ LayerTestResult SquareInt16Test( armnn::IWorkloadFactory& workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); +// +// HardSwish +// + +LayerTestResult HardSwishTest( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + +LayerTestResult HardSwishUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + +LayerTestResult HardSwishInt16Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager); + // // Other // -- cgit v1.2.1