aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h
diff options
context:
space:
mode:
authorSheri Zhang <sheri.zhang@arm.com>2020-04-21 13:10:24 +0100
committerSheri Zhang <sheri.zhang@arm.com>2020-04-26 21:31:26 +0000
commit3a35398ed6cc5d9c0f45f33dabb2bfbb017bcf60 (patch)
treeaaa4a8949e288157ab92404d1745214529c0c69b /arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h
parent31b49caa2ca9308a5ba62a598afc9d1982b4af18 (diff)
downloadComputeLibrary-3a35398ed6cc5d9c0f45f33dabb2bfbb017bcf60.tar.gz
COMPMID-3240: Add support for layer normalization to CLQLSTMLayer
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com> Change-Id: I45359a4ddb46c059097a2d77c008f802e8f4c143 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3065 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h')
-rw-r--r--arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h b/arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h
index 1a2f3111f5..2d4707245f 100644
--- a/arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h
+++ b/arm_compute/core/CL/kernels/CLQLSTMLayerNormalizationKernel.h
@@ -73,7 +73,7 @@ public:
*
* @return a status
*/
- static Status validate(const ITensorInfo *input, ITensorInfo *output, const ITensorInfo *weight, const ITensorInfo *bias);
+ static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *weight, const ITensorInfo *bias);
// Inherited methods overridden:
void run(const Window &window, cl::CommandQueue &queue) override;