diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-07-14 15:05:31 +0100 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-08-05 17:45:10 +0000 |
| commit | 4d81d75de09bba2de17e849a3d46e3052d449ad8 (patch) | |
| tree | d3bc5c3430e3d8cf70540be5e545914a54e93035 /src/core/CL/CLKernelLibrary.cpp | |
| parent | 801bbcb2f9a16633d0071b55aaf93b89870248ee (diff) | |
| download | ComputeLibrary-4d81d75de09bba2de17e849a3d46e3052d449ad8.tar.gz | |
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 <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3606
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Tested-by: Arm Jenkins <bsgcomp@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 | 5 |
1 files changed, 5 insertions, 0 deletions
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<std::string, std::string> 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" }, @@ -939,6 +940,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map = #include "./cl_kernels/types.hembed" }, { + "unpooling_layer.cl", +#include "./cl_kernels/unpooling_layer.clembed" + }, + { "warp_affine.cl", #include "./cl_kernels/warp_affine.clembed" }, |