diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-08-21 17:53:38 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 477531c258801caf3cce44eb3e43df611b42fc6d (patch) | |
| tree | 824294059aa01b658e80c760a5fb0773341add4f /src/core/CL/CLKernelLibrary.cpp | |
| parent | 4854fc3d9cac1de37e4670bc4272f44b43be8fa3 (diff) | |
| download | ComputeLibrary-477531c258801caf3cce44eb3e43df611b42fc6d.tar.gz | |
COMPMID-1527 - Implementing ReorgLayer on OpenCL
Also extended tests on NEON
Change-Id: Icb0eced534e904ef807972dd3a31988f501bb02e
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/147095
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/CL/CLKernelLibrary.cpp')
| -rw-r--r-- | src/core/CL/CLKernelLibrary.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/core/CL/CLKernelLibrary.cpp b/src/core/CL/CLKernelLibrary.cpp index 4af2b09530..7cc586bff1 100644 --- a/src/core/CL/CLKernelLibrary.cpp +++ b/src/core/CL/CLKernelLibrary.cpp @@ -336,6 +336,8 @@ const std::map<std::string, std::string> CLKernelLibrary::_kernel_program_map = { "reduction_operation", "reduction_operation.cl" }, { "remap_nearest_neighbour", "remap.cl" }, { "remap_bilinear", "remap.cl" }, + { "reorg_layer_nchw", "reorg_layer.cl" }, + { "reorg_layer_nhwc", "reorg_layer.cl" }, { "reshape_layer", "reshape_layer.cl" }, { "reshape_to_columns", "convolution_layer.cl" }, { "RGB888_to_IYUV_bt709", "color_convert.cl" }, @@ -720,6 +722,10 @@ const std::map<std::string, std::string> CLKernelLibrary::_program_source_map = #include "./cl_kernels/remap.clembed" }, { + "reorg_layer.cl", +#include "./cl_kernels/reorg_layer.clembed" + }, + { "reshape_layer.cl", #include "./cl_kernels/reshape_layer.clembed" }, |