* @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),
* @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),
* @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),