diff options
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 1c773bc42f..577ba762cb 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -313,8 +313,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "pooling_layer_3", "pooling_layer.cl" }, { "pooling_layer_optimized_3", "pooling_layer.cl" }, { "pooling_layer_7", "pooling_layer.cl" }, - { "pooling_layer_MxN", "pooling_layer.cl" }, - { "pooling_layer_MxN_quantized", "pooling_layer_quantized.cl" }, + { "pooling_layer_MxN_nchw", "pooling_layer.cl" }, + { "pooling_layer_MxN_nhwc", "pooling_layer.cl" }, + { "pooling_layer_MxN_quantized_nhwc", "pooling_layer_quantized.cl" }, + { "pooling_layer_MxN_quantized_nchw", "pooling_layer_quantized.cl" }, { "quantization_layer", "quantization_layer.cl" }, { "reduction_operation", "reduction_operation.cl" }, { "remap_nearest_neighbour", "remap.cl" }, |