diff options
Diffstat (limited to 'src/gpu/cl/ClKernelLibrary.cpp')
-rw-r--r-- | src/gpu/cl/ClKernelLibrary.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/gpu/cl/ClKernelLibrary.cpp b/src/gpu/cl/ClKernelLibrary.cpp index f675dbe6fd..1bf7f2b3ac 100644 --- a/src/gpu/cl/ClKernelLibrary.cpp +++ b/src/gpu/cl/ClKernelLibrary.cpp @@ -438,6 +438,7 @@ const std::map<std::string, std::string> ClKernelLibrary::_kernel_program_map = { "pooling_layer_2x2_nhwc", "nhwc/pooling_layer.cl" }, { "pooling_layer_MxN_quantized_nhwc", "nhwc/pooling_layer_quantized.cl" }, { "pooling_3d_layer_MxN_ndhwc", "nhwc/pooling_3d_layer.cl" }, + { "pooling_3d_layer_MxN_ndhwc_quantized", "nhwc/pooling_3d_layer_quantized.cl" }, { "reorg_layer_nhwc", "nhwc/reorg_layer.cl" }, { "scale_nearest_neighbour_nhwc", "nhwc/scale.cl" }, { "scale_bilinear_nhwc", "nhwc/scale.cl" }, @@ -907,6 +908,10 @@ const std::map<std::string, std::string> ClKernelLibrary::_program_source_map = #include "./cl_kernels/nhwc/pooling_3d_layer.clembed" }, { + "nhwc/pooling_3d_layer_quantized.cl", +#include "./cl_kernels/nhwc/pooling_3d_layer_quantized.clembed" + }, + { "nhwc/pooling_layer_quantized.cl", #include "./cl_kernels/nhwc/pooling_layer_quantized.clembed" }, |