diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-07 11:04:57 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-04-14 15:13:37 +0000 |
commit | b18252dce941001d8980721596709ea01d55747a (patch) | |
tree | 013d187c9bb5ab0704dd4e7c5f3f08e030d752b8 /src/core/CL/CLKernelLibrary.cpp | |
parent | 5a4fe19c23729f1e58e947ed15e865dc33c35ff6 (diff) | |
download | ComputeLibrary-b18252dce941001d8980721596709ea01d55747a.tar.gz |
COMPMID-3239: Implement QSYMM16 LayerNormalizationKernel for CL
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: Ib1577c4a9aa29293a903731b2a4083b0d2243e1e
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2994
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
-rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 7437f1bf22..00e7b2bc5c 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -326,6 +326,7 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "pooling_layer_MxN_quantized_nhwc", "pooling_layer_quantized.cl" }, { "pooling_layer_MxN_quantized_nchw", "pooling_layer_quantized.cl" }, { "prior_box_layer_nchw", "prior_box_layer.cl" }, + { "qlstm_layer_normalization", "qlstm_layer_normalization.cl" }, { "quantization_layer", "quantization_layer.cl" }, { "range", "range.cl" }, { "range_quantized", "range.cl" }, @@ -824,6 +825,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map = #include "./cl_kernels/prior_box_layer.clembed" }, { + "qlstm_layer_normalization.cl", +#include "./cl_kernels/qlstm_layer_normalization.clembed" + }, + { "quantization_layer.cl", #include "./cl_kernels/quantization_layer.clembed" }, |