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/RefLayerSupport.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/backends/reference/RefLayerSupport.cpp') diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 25334c3b52..7d5c3b509e 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -109,6 +109,7 @@ bool RefLayerSupport::IsActivationSupported(const TensorInfo& input, case ActivationFunction::Abs: case ActivationFunction::BoundedReLu: case ActivationFunction::Elu: + case ActivationFunction::HardSwish: case ActivationFunction::LeakyReLu: case ActivationFunction::Linear: case ActivationFunction::ReLu: -- cgit v1.2.1