diff options
author | Usama Arif <usama.arif@arm.com> | 2019-05-14 17:48:47 +0100 |
---|---|---|
committer | Usama Arif <usama.arif@arm.com> | 2019-05-16 12:12:32 +0000 |
commit | 80e55db85bb4842ef287c2732fc98ad869748b0a (patch) | |
tree | c50116c531d6f10ca2f610d03ddb6e84f7209a94 /tests | |
parent | 0681e3bf3b2abf9a0704c3243859a60204d3565c (diff) | |
download | ComputeLibrary-80e55db85bb4842ef287c2732fc98ad869748b0a.tar.gz |
COMPMID-2194: Add identity function to NEActivationLayerKernel.cpp
Change-Id: Id9bc53722b7dfe277812e9e1d749ac4cea35260a
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/184712
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1143
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'tests')
-rw-r--r-- | tests/datasets/ActivationFunctionsDataset.h | 3 | ||||
-rw-r--r-- | tests/validation/reference/ActivationLayer.h | 3 |
2 files changed, 5 insertions, 1 deletions
diff --git a/tests/datasets/ActivationFunctionsDataset.h b/tests/datasets/ActivationFunctionsDataset.h index 147c5ae51b..c5dc28f0c3 100644 --- a/tests/datasets/ActivationFunctionsDataset.h +++ b/tests/datasets/ActivationFunctionsDataset.h @@ -51,7 +51,8 @@ public: ActivationLayerInfo::ActivationFunction::SOFT_RELU, ActivationLayerInfo::ActivationFunction::SQRT, ActivationLayerInfo::ActivationFunction::SQUARE, - ActivationLayerInfo::ActivationFunction::TANH + ActivationLayerInfo::ActivationFunction::TANH, + ActivationLayerInfo::ActivationFunction::IDENTITY }) { } diff --git a/tests/validation/reference/ActivationLayer.h b/tests/validation/reference/ActivationLayer.h index 77b353056e..cc43bc10a8 100644 --- a/tests/validation/reference/ActivationLayer.h +++ b/tests/validation/reference/ActivationLayer.h @@ -75,6 +75,9 @@ inline T activate_float(T x, T a, T b, ActivationLayerInfo::ActivationFunction a case ActivationLayerInfo::ActivationFunction::TANH: ret = a * std::tanh(b * x); break; + case ActivationLayerInfo::ActivationFunction::IDENTITY: + ret = x; + break; default: ARM_COMPUTE_ERROR("Unsupported activation function"); break; |