diff options
| author | Manuel Bottini <manuel.bottini@arm.com> | 2020-04-21 13:29:30 +0100 |
|---|---|---|
| committer | Manuel Bottini <manuel.bottini@arm.com> | 2020-04-22 08:57:15 +0000 |
| commit | 256c0b9d5d66d35d52c7eee3599a4d91e7887ec8 (patch) | |
| tree | 6bb2bd6648ed3a7292d63c731a679bb7cb2cf3d3 /arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h | |
| parent | 7ba240bb54bdbd50f1809f944dde2fac7005cc10 (diff) | |
| download | ComputeLibrary-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/CLFuseBatchNormalizationKernel.h')
| -rw-r--r-- | arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h b/arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h index 2d62a576bb..2fe6b223ca 100644 --- a/arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h +++ b/arm_compute/core/CL/kernels/CLFuseBatchNormalizationKernel.h @@ -81,7 +81,7 @@ public: * @param[in] epsilon (Optional) Batch normalization layer epsilon parameter. Defaults to 0.001f. * @param[in] fbn_type (Optional) Fused batch normalization type. Defaults to CONVOLUTION. */ - void configure(CLCompileContext &compile_context, const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_weights, ICLTensor *fused_bias, + void configure(const CLCompileContext &compile_context, const ICLTensor *input_weights, const ICLTensor *bn_mean, const ICLTensor *bn_var, ICLTensor *fused_weights, ICLTensor *fused_bias, const ICLTensor *input_bias = nullptr, const ICLTensor *bn_beta = nullptr, const ICLTensor *bn_gamma = nullptr, float epsilon = 0.001f, FuseBatchNormalizationType fbn_type = FuseBatchNormalizationType::CONVOLUTION); /** Static function to check if given info will lead to a valid configuration of @ref CLFuseBatchNormalizationKernel |