diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-01-29 15:05:41 +0000 |
---|---|---|
committer | Isabella Gottardi <isabella.gottardi@arm.com> | 2019-01-30 10:22:24 +0000 |
commit | 5d62c0117e1dc8db953e3e8c4b3fa50c6bbf4536 (patch) | |
tree | 2a7e76d694aeeff39f94c6205726db26e252a208 | |
parent | 83bbf77358fa682e426b80d92de5a0c1ba05347d (diff) | |
download | ComputeLibrary-5d62c0117e1dc8db953e3e8c4b3fa50c6bbf4536.tar.gz |
COMPMID-1710: Fix NEActivationLayer map for QASYMM8
Change-Id: I6c11568f8933d074541a28ef782954460e00d851
Signed-off-by: Isabella Gottardi <isabella.gottardi@arm.com>
Reviewed-on: https://review.mlplatform.org/596
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
-rw-r--r-- | src/core/NEON/kernels/NEActivationLayerKernel.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEActivationLayerKernel.cpp b/src/core/NEON/kernels/NEActivationLayerKernel.cpp index 7565a1214c..c61d851ad8 100644 --- a/src/core/NEON/kernels/NEActivationLayerKernel.cpp +++ b/src/core/NEON/kernels/NEActivationLayerKernel.cpp @@ -140,6 +140,7 @@ void NEActivationLayerKernel::configure(ITensor *input, ITensor *output, Activat // Activation functions : QASYMM8 static std::map<ActivationFunction, ActivationFunctionExecutorPtr> act_map_qasymm8 = { + { ActivationFunction::BOUNDED_RELU, &NEActivationLayerKernel::activation<ActivationFunction::BOUNDED_RELU, qasymm8_t> }, { ActivationFunction::LU_BOUNDED_RELU, &NEActivationLayerKernel::activation<ActivationFunction::LU_BOUNDED_RELU, qasymm8_t> }, { ActivationFunction::RELU, &NEActivationLayerKernel::activation<ActivationFunction::RELU, qasymm8_t> }, }; |