diff options
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"); |