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 --- src/backends/reference/test/RefLayerTests.cpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/backends/reference/test') diff --git a/src/backends/reference/test/RefLayerTests.cpp b/src/backends/reference/test/RefLayerTests.cpp index ed2b995bd5..40bf600331 100644 --- a/src/backends/reference/test/RefLayerTests.cpp +++ b/src/backends/reference/test/RefLayerTests.cpp @@ -466,6 +466,10 @@ ARMNN_AUTO_TEST_CASE(TanhInt16, TanhInt16Test) ARMNN_AUTO_TEST_CASE(Elu, EluTest) ARMNN_AUTO_TEST_CASE(EluUint8, EluUint8Test) ARMNN_AUTO_TEST_CASE(EluInt16, EluInt16Test) +// HardSwish Activation +ARMNN_AUTO_TEST_CASE(HardSwish, HardSwishTest) +ARMNN_AUTO_TEST_CASE(HardSwishUint8, HardSwishUint8Test) +ARMNN_AUTO_TEST_CASE(HardSwishInt16, HardSwishInt16Test) // Fully Connected ARMNN_AUTO_TEST_CASE(SimpleFullyConnected, FullyConnectedFloat32Test, false, false) -- cgit v1.2.1