aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/CL/kernels/CLPoolingLayerKernel.h')
-rw-r--r--arm_compute/core/CL/kernels/CLPoolingLayerKernel.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h
index 6c5091ff9e..971e1506af 100644
--- a/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h
+++ b/arm_compute/core/CL/kernels/CLPoolingLayerKernel.h
@@ -49,7 +49,7 @@ public:
/** Set the input and output tensors.
*
- * @param[in] input Source tensor. Data types supported: F16, F32.
+ * @param[in] input Source tensor. Data types supported: F16/F32.
* @param[out] output Destination tensor. Data types supported: Same as @p input.
* @param[in] pool_info Contains pooling operation information described in @ref PoolingLayerInfo.
* Supported pooling sizes : 2, 3 and 7
@@ -65,6 +65,7 @@ private:
ICLTensor *_output;
PoolingLayerInfo _pool_info;
BorderSize _border_size;
+ unsigned int _num_elems_processed_per_iteration;
};
}
#endif /*__ARM_COMPUTE_CLPOOLINGLAYERKERNEL_H__ */