diff options
Diffstat (limited to 'tests/datasets/ActivationFunctionsDataset.h')
-rw-r--r-- | tests/datasets/ActivationFunctionsDataset.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/datasets/ActivationFunctionsDataset.h b/tests/datasets/ActivationFunctionsDataset.h index 85febd4df4..9b0d775376 100644 --- a/tests/datasets/ActivationFunctionsDataset.h +++ b/tests/datasets/ActivationFunctionsDataset.h @@ -54,7 +54,9 @@ public: ActivationLayerInfo::ActivationFunction::SQUARE, ActivationLayerInfo::ActivationFunction::TANH, ActivationLayerInfo::ActivationFunction::IDENTITY, +#ifdef __aarch64__ ActivationLayerInfo::ActivationFunction::GELU, +#endif /* __aarch64__ */ }) { } |