From a83af7b388a1ac721c6a85b0e80ac2950687dfad Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Wed, 18 Mar 2020 15:58:11 +0000 Subject: IVGCVSW-4446 Add HardSwish EndToEnd tests * adds convert from armnn HardSwish to Acl HardSwish * adds EndToEnd tests for Ref, Cl and Neon Signed-off-by: Jan Eilers Change-Id: If997dad783ff45d7f061254e7e30bb69d4d4d564 --- src/backends/aclCommon/ArmComputeUtils.hpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/backends/aclCommon') diff --git a/src/backends/aclCommon/ArmComputeUtils.hpp b/src/backends/aclCommon/ArmComputeUtils.hpp index 01a5445844..c3cfb5cb78 100644 --- a/src/backends/aclCommon/ArmComputeUtils.hpp +++ b/src/backends/aclCommon/ArmComputeUtils.hpp @@ -66,6 +66,7 @@ ConvertActivationFunctionToAclActivationFunction(ActivationFunction armnnFunctio case ActivationFunction::Square: return AclActivationFunction::SQUARE; case ActivationFunction::TanH: return AclActivationFunction::TANH; case ActivationFunction::Elu: return AclActivationFunction::ELU; + case ActivationFunction::HardSwish: return AclActivationFunction::HARD_SWISH; default: throw InvalidArgumentException("Unsupported activation function"); } } -- cgit v1.2.1