diff options
author | Jan Eilers <jan.eilers@arm.com> | 2020-03-18 15:58:11 +0000 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2020-03-18 18:33:38 +0000 |
commit | a83af7b388a1ac721c6a85b0e80ac2950687dfad (patch) | |
tree | 0423109f7269ae66f2bfceaea6c246312712ae15 /src/backends/aclCommon | |
parent | bca73e1c82438f160364a113793d0a2195c665ac (diff) | |
download | armnn-a83af7b388a1ac721c6a85b0e80ac2950687dfad.tar.gz |
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 <jan.eilers@arm.com>
Change-Id: If997dad783ff45d7f061254e7e30bb69d4d4d564
Diffstat (limited to 'src/backends/aclCommon')
-rw-r--r-- | src/backends/aclCommon/ArmComputeUtils.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
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"); } } |