diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-06-27 17:39:11 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:15:39 +0100 |
| commit | ccc65d44a53eaa61c718cbc4d826c811e2ccebda (patch) | |
| tree | a5394c05c43ece384d629ceae644f5c7b5d23280 /tests/validation/TensorOperations.h | |
| parent | 0745a980c6a5e2d294bcd09f3c704e6cf4fe316d (diff) | |
| download | ComputeLibrary-ccc65d44a53eaa61c718cbc4d826c811e2ccebda.tar.gz | |
COMPMID-427: Port NEActivationLayer in 16bit fixed point.
Change-Id: Iebd61807f7b597c6bd990673bc7655c68ee16f4b
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79085
Reviewed-by: Moritz Pflanzer <moritz.pflanzer@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Diffstat (limited to 'tests/validation/TensorOperations.h')
| -rw-r--r-- | tests/validation/TensorOperations.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/TensorOperations.h b/tests/validation/TensorOperations.h index 0d752ee6fc..adac70901d 100644 --- a/tests/validation/TensorOperations.h +++ b/tests/validation/TensorOperations.h @@ -930,7 +930,7 @@ void activation_layer(const Tensor<T> &in, Tensor<T> &out, ActivationLayerInfo a out[i] = mul(x, x).raw(); break; case ActivationLayerInfo::ActivationFunction::TANH: - out[i] = tanh(x).raw(); + out[i] = mul(a, tanh(mul(b, x))).raw(); break; default: ARM_COMPUTE_ERROR("Activation function not recognised"); |