commit 7e9391bb14d219cda310bff355669b5964b1f576
author Michalis Spyrou <michalis.spyrou@arm.com> Fri Oct 05 14:49:28 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:55:45 2018 +0000
tree 789142f63d4c8e95612b042d07a0683cfe596fb9
parent 555c3d6448a1dc7b326fad2ab7f75eccc8e5cff6

COMPMID-1574 Implement ReduceMean in OpenCL

Change-Id: Id331199f569f52a37280a9ada5bf84694580b93c
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/152843
Tested-by: bsgcomp <bsgcomp@arm.com>
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>

arm_compute/runtime/CL/functions/CLReductionOperation.h

diff --git a/arm_compute/runtime/CL/functions/CLReductionOperation.h b/arm_compute/runtime/CL/functions/CLReductionOperation.h
index d862aff..4208178 100644
--- a/arm_compute/runtime/CL/functions/CLReductionOperation.h
+++ b/arm_compute/runtime/CL/functions/CLReductionOperation.h
@@ -53,7 +53,7 @@
 
     /** Set the input and output tensors.
      *
-     * @param[in]  input  Source tensor. Data types supported: F16/F32. Data layouts supported: NCHW.
+     * @param[in]  input  Source tensor. Data types supported: QASYMM8/F16/F32. Data layouts supported: NCHW.
      * @param[out] output Destination tensor. Data types and data layouts supported: Same as @p input.
      * @param[in]  axis   Axis along which to reduce. Supported reduction axis : 0
      * @param[in]  op     Reduction operation to perform.
@@ -62,7 +62,7 @@
 
     /** Static function to check if given info will lead to a valid configuration of @ref CLReductionOperation.
      *
-     * @param[in] input  Source tensor info. Data types supported: F16/F32. Data layouts supported: NCHW.
+     * @param[in] input  Source tensor info. Data types supported: QASYMM8/F16/F32. Data layouts supported: NCHW.
      * @param[in] output Destination tensor info. Data types and data layouts supported: Same as @p input.
      * @param[in] axis   Axis along which to reduce. Supported reduction axis : 0
      * @param[in] op     Reduction operation to perform.
@@ -80,6 +80,8 @@
     std::unique_ptr<CLReductionOperationKernel[]> _reduction_kernels_vector{ nullptr };
     std::unique_ptr<CLFillBorderKernel[]>         _border_handlers_vector{ nullptr };
     unsigned int                                  _num_of_stages;
+    unsigned int                                  _reduction_axis;
+    bool                                          _is_quantized;
 };
 }
 #endif /*__ARM_COMPUTE_CLREDUCTIONOPERATION_H__ */