diff options
author | Frederick Liardet <frederick.liardet@arm.com> | 2021-04-22 21:13:21 +0100 |
---|---|---|
committer | frederick.liardet <frederick.liardet@arm.com> | 2021-06-15 11:24:53 +0000 |
commit | 36dff9f81e3a95aea19fcc7246a4896930a14bc6 (patch) | |
tree | 64f3194e806bb4a8a5e6f2f30c202295c5e853c6 /src/core/gpu/cl | |
parent | ee301b384f4aeb697a5c249b8bb848d784146582 (diff) | |
download | ComputeLibrary-36dff9f81e3a95aea19fcc7246a4896930a14bc6.tar.gz |
Add NHWC support to CLRemap
Add NHWC support to CLRemap, also add relevant tests.
Partially resolves COMPMID-4335.
Change-Id: I119bea99be497fb85d5cd83a10f8d4e8e1f97f17
Signed-off-by: Freddie Liardet <frederick.liardet@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5773
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/gpu/cl')
-rw-r--r-- | src/core/gpu/cl/ClKernelLibrary.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/core/gpu/cl/ClKernelLibrary.cpp b/src/core/gpu/cl/ClKernelLibrary.cpp index b0458d7c3a..9d516e54a7 100644 --- a/src/core/gpu/cl/ClKernelLibrary.cpp +++ b/src/core/gpu/cl/ClKernelLibrary.cpp @@ -397,8 +397,10 @@ const std::map<std::string, std::string> ClKernelLibrary::_kernel_program_map = { "reduction_operation_y", "reduction_operation.cl" }, { "reduction_operation_z", "reduction_operation.cl" }, { "reduction_operation_w", "reduction_operation.cl" }, - { "remap_nearest_neighbour", "remap.cl" }, - { "remap_bilinear", "remap.cl" }, + { "remap_nearest_neighbour_nchw", "remap.cl" }, + { "remap_bilinear_nchw", "remap.cl" }, + { "remap_nearest_neighbour_nhwc", "remap.cl" }, + { "remap_bilinear_nhwc", "remap.cl" }, { "reorg_layer_nchw", "reorg_layer.cl" }, { "reorg_layer_nhwc", "reorg_layer.cl" }, { "reshape_layer", "reshape_layer.cl" }, |