diff options
Diffstat (limited to 'tests/datasets/ActivationFunctionsDataset.h')
-rw-r--r-- | tests/datasets/ActivationFunctionsDataset.h | 3 |
1 files changed, 2 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 }) { } |