diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-07-12 16:12:12 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
| commit | 579c0498e161215be1a36080b0b454e5198a992a (patch) | |
| tree | 1ec07b602935e7261a8a7aea900dc925e9bc35a1 /src/core/Utils.cpp | |
| parent | 81f0d15d6840a0ae8ef571114555a26da74c4a43 (diff) | |
| download | ComputeLibrary-579c0498e161215be1a36080b0b454e5198a992a.tar.gz | |
COMPMID-417: Add Leaky RELU support for both NEON/CL.
-Adds parametrizable leaky relu (x>0) ? x : a*x.
Change-Id: Ief19a435b5832a30b56f4aaaf55125787addee94
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/80575
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'src/core/Utils.cpp')
| -rw-r--r-- | src/core/Utils.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index 11b41aa178..cf8e1940ec 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -156,6 +156,7 @@ const std::string &arm_compute::string_from_activation_func(ActivationLayerInfo: { ActivationLayerInfo::ActivationFunction::LOGISTIC, "LOGISTIC" }, { ActivationLayerInfo::ActivationFunction::RELU, "RELU" }, { ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, "BRELU" }, + { ActivationLayerInfo::ActivationFunction::LEAKY_RELU, "LRELU" }, { ActivationLayerInfo::ActivationFunction::SOFT_RELU, "SRELU" }, { ActivationLayerInfo::ActivationFunction::SQRT, "SQRT" }, { ActivationLayerInfo::ActivationFunction::SQUARE, "SQUARE" }, |