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/CLScharr3x3Kernel.cpp b/src/core/CL/kernels/CLScharr3x3Kernel.cpp
index 5182390..1c1fedc 100644
--- a/src/core/CL/kernels/CLScharr3x3Kernel.cpp
+++ b/src/core/CL/kernels/CLScharr3x3Kernel.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016-2018 ARM Limited.
+ * Copyright (c) 2016-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -115,16 +115,8 @@
{
unsigned int idx = 0;
add_2D_tensor_argument(idx, _input, slice);
-
- if(_run_scharr_x)
- {
- add_2D_tensor_argument(idx, _output_x, slice);
- }
-
- if(_run_scharr_y)
- {
- add_2D_tensor_argument(idx, _output_y, slice);
- }
+ add_2D_tensor_argument_if((_run_scharr_x), idx, _output_x, slice);
+ add_2D_tensor_argument_if((_run_scharr_y), idx, _output_y, slice);
enqueue(queue, *this, slice);
}