From e1651a55c637708347c9adda4a77f1deffacf7d2 Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 11 Jul 2019 15:00:49 +0100 Subject: COMPMID-2414 Add add_2D_tensor_argument_if for OpenCL Change-Id: I8675b4b4face0be9416ae6c7a7023234d50fc0cb Signed-off-by: Michalis Spyrou Reviewed-on: https://review.mlplatform.org/c/1524 Comments-Addressed: Arm Jenkins Reviewed-by: Michele Di Giorgio Tested-by: Arm Jenkins --- src/core/CL/kernels/CLDerivativeKernel.cpp | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) (limited to 'src/core/CL/kernels/CLDerivativeKernel.cpp') diff --git a/src/core/CL/kernels/CLDerivativeKernel.cpp b/src/core/CL/kernels/CLDerivativeKernel.cpp index 0a247f9071..e6d694d875 100644 --- a/src/core/CL/kernels/CLDerivativeKernel.cpp +++ b/src/core/CL/kernels/CLDerivativeKernel.cpp @@ -142,16 +142,8 @@ void CLDerivativeKernel::run(const Window &window, cl::CommandQueue &queue) { unsigned int idx = 0; add_2D_tensor_argument(idx, _input, slice); - - if(_run_derivative_x) - { - add_2D_tensor_argument(idx, _output_x, slice); - } - - if(_run_derivative_y) - { - add_2D_tensor_argument(idx, _output_y, slice); - } + add_2D_tensor_argument_if((_run_derivative_x), idx, _output_x, slice); + add_2D_tensor_argument_if((_run_derivative_y), idx, _output_y, slice); enqueue(queue, *this, slice, lws_hint()); } -- cgit v1.2.1