From 36dff9f81e3a95aea19fcc7246a4896930a14bc6 Mon Sep 17 00:00:00 2001
From: Frederick Liardet <frederick.liardet@arm.com>
Date: Thu, 22 Apr 2021 21:13:21 +0100
Subject: 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>
---
 src/core/gpu/cl/ClKernelLibrary.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'src/core/gpu/cl')

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" },
-- 
cgit v1.2.1