diff options
Diffstat (limited to 'tests/validation/reference/ActivationLayer.h')
| -rw-r--r-- | tests/validation/reference/ActivationLayer.h | 3 |
1 files changed, 3 insertions, 0 deletions
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; |