aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/TensorOperations.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/TensorOperations.h')
-rw-r--r--tests/validation/TensorOperations.h2
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");