diff options
| author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-03-13 14:56:05 +0000 |
|---|---|---|
| committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-04-07 09:00:09 +0000 |
| commit | 0d008f77b0085619c446d0ab5dc1228a80776706 (patch) | |
| tree | e1f6e91bf8da63e8ef98e11ab8eb6a6972a284f2 /arm_compute/core/NEON/NEKernels.h | |
| parent | 4df2cf3177129d10500d30056bf8404418f703d6 (diff) | |
| download | ComputeLibrary-0d008f77b0085619c446d0ab5dc1228a80776706.tar.gz | |
COMPMID-3281: Implement QSYMM16 Layer Normalization for NEON QLSTM
- Reference kernel is modified to use the same algorithm as NEON kernel.
- NEON kernel is implemented.
- Tests for validation and run are added.
Change-Id: I3533bc2bd12c6e9cc75d837ecf193f74ceddf796
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2948
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/NEKernels.h')
| -rw-r--r-- | arm_compute/core/NEON/NEKernels.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/NEKernels.h b/arm_compute/core/NEON/NEKernels.h index d9f8f00c0b..38701f434a 100644 --- a/arm_compute/core/NEON/NEKernels.h +++ b/arm_compute/core/NEON/NEKernels.h @@ -120,6 +120,7 @@ #include "arm_compute/core/NEON/kernels/NEPixelWiseMultiplicationKernel.h" #include "arm_compute/core/NEON/kernels/NEPoolingLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEPriorBoxLayerKernel.h" +#include "arm_compute/core/NEON/kernels/NEQLSTMLayerNormalizationKernel.h" #include "arm_compute/core/NEON/kernels/NEQuantizationLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEROIAlignLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEROIPoolingLayerKernel.h" |