diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2020-08-03 20:11:56 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-08-05 15:59:59 +0000 |
| commit | 801bbcb2f9a16633d0071b55aaf93b89870248ee (patch) | |
| tree | 4581fd059191e89803f3d77db0b052714bc20eba /src/core/CL/CLKernelLibrary.cpp | |
| parent | 0499dff9293a86d3d53f72fed0a38b2823563674 (diff) | |
| download | ComputeLibrary-801bbcb2f9a16633d0071b55aaf93b89870248ee.tar.gz | |
COMPMID-2479: Extend CLPoolingLayer max pooling to extract indices
Fix PoolingLayer max pooling reference bug to extract indices.
Extend CLPoolingLayer max pooling to extract indices, all the paddings need to be substracted.
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: If8e82e7f7e03172ad05f5a7cd5f13cf682fd1ffc
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3649
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 73874b69b2..9dec79b01a 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -325,6 +325,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "pooling_layer_7", "pooling_layer.cl" }, { "pooling_layer_MxN_nchw", "pooling_layer.cl" }, { "pooling_layer_MxN_nhwc", "pooling_layer.cl" }, + { "pooling_layer_2_nhwc_indices_fp32", "pooling_layer.cl" }, + { "pooling_layer_2_nhwc_indices_fp16", "pooling_layer.cl" }, + { "pooling_layer_2_nchw_indices_fp32", "pooling_layer.cl" }, + { "pooling_layer_2_nchw_indices_fp16", "pooling_layer.cl" }, { "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" }, |