COMPMID-2414 Add add_2D_tensor_argument_if for OpenCL
Change-Id: I8675b4b4face0be9416ae6c7a7023234d50fc0cb
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1524
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/core/CL/kernels/CLChannelCombineKernel.cpp b/src/core/CL/kernels/CLChannelCombineKernel.cpp
index 11cd8ec..126614a 100644
--- a/src/core/CL/kernels/CLChannelCombineKernel.cpp
+++ b/src/core/CL/kernels/CLChannelCombineKernel.cpp
@@ -262,11 +262,7 @@
add_2D_tensor_argument(idx, _planes[0], slice);
add_2D_tensor_argument(idx, _planes[1], win_sub_plane1);
add_2D_tensor_argument(idx, _planes[2], win_sub_plane2);
-
- if(nullptr != _planes[3])
- {
- add_2D_tensor_argument(idx, _planes[3], slice);
- }
+ add_2D_tensor_argument_if((nullptr != _planes[3]), idx, _planes[3], slice);
// Set outputs
if(nullptr != _output) // Single planar output
@@ -280,11 +276,7 @@
add_2D_tensor_argument(idx, _output_multi->cl_plane(0), slice);
add_2D_tensor_argument(idx, _output_multi->cl_plane(1), win_sub_plane1);
-
- if(3 == num_planes_from_format(_output_multi->info()->format()))
- {
- add_2D_tensor_argument(idx, _output_multi->cl_plane(2), win_sub_plane2);
- }
+ add_2D_tensor_argument_if((3 == num_planes_from_format(_output_multi->info()->format())), idx, _output_multi->cl_plane(2), win_sub_plane2);
_kernel.setArg(idx++, slice.y().end());
}