diff options
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" }, |