aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2020-04-21 13:29:30 +0100
committerManuel Bottini <manuel.bottini@arm.com>2020-04-22 08:57:15 +0000
commit256c0b9d5d66d35d52c7eee3599a4d91e7887ec8 (patch)
tree6bb2bd6648ed3a7292d63c731a679bb7cb2cf3d3 /arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h
parent7ba240bb54bdbd50f1809f944dde2fac7005cc10 (diff)
downloadComputeLibrary-256c0b9d5d66d35d52c7eee3599a4d91e7887ec8.tar.gz
COMPMID-3280: Make all ML primitives for CL use the new interface - Part1 - Fix1
- const fix in the CLKernels part 1 Change-Id: I17340cb6ff26afd52b14b46645efedbe07aef1b6 Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3067 Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h')
-rw-r--r--arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h5
1 files changed, 2 insertions, 3 deletions
diff --git a/arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h b/arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h
index 564b21680b..8eaaca845a 100644
--- a/arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h
+++ b/arm_compute/core/CL/kernels/CLBatchNormalizationLayerKernel.h
@@ -81,9 +81,8 @@ public:
* @param[in] epsilon (Optional) Small value to avoid division with zero. Default value is 0.001f.
* @param[in] act_info (Optional) Activation layer information in case of a fused activation. Only RELU, BOUNDED_RELU and LU_BOUNDED_RELU supported.
*/
- void configure(CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output, const ICLTensor *mean, const ICLTensor *var, const ICLTensor *beta = nullptr, const ICLTensor *gamma = nullptr,
- float epsilon = 0.001f,
- ActivationLayerInfo act_info = ActivationLayerInfo());
+ void configure(const CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output, const ICLTensor *mean, const ICLTensor *var, const ICLTensor *beta = nullptr,
+ const ICLTensor *gamma = nullptr, float epsilon = 0.001f, ActivationLayerInfo act_info = ActivationLayerInfo());
/** Static function to check if given info will lead to a valid configuration of @ref CLBatchNormalizationLayerKernel
*
* @param[in] input Source tensor info. In case of @p output tensor info = nullptr, this tensor will store the result.