Redefining input parameters in reduce_operations.cl Kernel (#3789)
authorPrasanna R/SNAP /SRI-Bangalore/Engineer/삼성전자 <prasanna.r@samsung.com>
Tue, 4 Dec 2018 00:46:05 +0000 (06:16 +0530)
committer오형석/동작제어Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Tue, 4 Dec 2018 00:46:05 +0000 (09:46 +0900)
This patch redefines input parameters in reduce_operations.cl Kernel.

Signed-off-by: prasannar <prasanna.r@samsung.com>
libs/ARMComputeEx/src/core/CL/cl_kernels/reduce_operation.cl

index 690ce7d..b8b80d7 100644 (file)
@@ -42,8 +42,8 @@
  * @param[in]  output_stride_w                      Stride of the source tensor in W dimension (in bytes)
  * @param[in]  output_step_w                        output_stride_w * number of elements along W processed per workitem(in bytes)
  * @param[in]  output_offset_first_element_in_bytes The offset of the first element in the destination image
- * @param[in]  stops                                Index of the last input element for each axis
- *                                                  If value is -1, that means the axis is reduced
+ * @param[in]  axis                                 Axis through which reduction occurs
+ * @param[in]  dim                                  Dimension across the axis to be reduced.
  */
 __kernel void reduce_max(TENSOR4D_DECLARATION(input),
                          TENSOR4D_DECLARATION(output),
@@ -150,8 +150,8 @@ __kernel void reduce_min(TENSOR4D_DECLARATION(input),
  * @param[in]  output_stride_w                      Stride of the source tensor in W dimension (in bytes)
  * @param[in]  output_step_w                        output_stride_w * number of elements along W processed per workitem(in bytes)
  * @param[in]  output_offset_first_element_in_bytes The offset of the first element in the destination image
- * @param[in]  stops                                Index of the last input element for each axis
- *                                                  If value is -1, that means the axis is reduced
+ * @param[in]  axis                                 Axis through which reduction occurs
+ * @param[in]  dim                                  Dimension across the axis to be reduced.
  */
 __kernel void reduce_mean(TENSOR4D_DECLARATION(input),
                           TENSOR4D_DECLARATION(output),
@@ -204,8 +204,8 @@ __kernel void reduce_mean(TENSOR4D_DECLARATION(input),
  * @param[in]  output_stride_w                      Stride of the source tensor in W dimension (in bytes)
  * @param[in]  output_step_w                        output_stride_w * number of elements along W processed per workitem(in bytes)
  * @param[in]  output_offset_first_element_in_bytes The offset of the first element in the destination image
- * @param[in]  stops                                Index of the last input element for each axis
- *                                                  If value is -1, that means the axis is reduced
+ * @param[in]  axis                                 Axis through which reduction occurs
+ * @param[in]  dim                                  Dimension across the axis to be reduced.
  */
 __kernel void reduce_sum(TENSOR4D_DECLARATION(input),
                          TENSOR4D_DECLARATION(output),