From 4d81d75de09bba2de17e849a3d46e3052d449ad8 Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Tue, 14 Jul 2020 15:05:31 +0100 Subject: COMPMID-2450: Implement CLMaxUnpoolingLayer - Add OpenCL kernel for Max unpooling layer - Add tests for validating the result Change-Id: If7ca79566a1198e3141f880abf46738980a62c81 Signed-off-by: Gian Marco Iodice Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3606 Reviewed-by: Pablo Marquez Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- src/core/CL/CLKernelLibrary.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/core/CL/CLKernelLibrary.cpp') diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 9dec79b01a..0b59ec8a71 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -283,6 +283,7 @@ const std::map CLKernelLibrary::_kernel_program_map = { "lktracker_stage0", "optical_flow_pyramid_lk.cl" }, { "lktracker_stage1", "optical_flow_pyramid_lk.cl" }, { "magnitude_phase", "magnitude_phase.cl" }, + { "max_unpooling_layer_2", "unpooling_layer.cl" }, { "mean_stddev_accumulate", "mean_stddev.cl" }, { "mean_stddev_normalization", "mean_stddev_normalization.cl" }, { "memset", "memset.cl" }, @@ -937,6 +938,10 @@ const std::map CLKernelLibrary::_program_source_map = { "types.h", #include "./cl_kernels/types.hembed" + }, + { + "unpooling_layer.cl", +#include "./cl_kernels/unpooling_layer.clembed" }, { "warp_affine.cl", -- cgit v1.2.1