diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-06-18 14:10:48 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-06-18 15:43:18 +0000 |
| commit | 51403b5b8ae92782e2ed467cae4c0783faa5a22b (patch) | |
| tree | 1a331a215df09a44056f4137e4fae7143b3477e4 /src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp | |
| parent | 917539225a4283c869b3cb2c82bdc94b5d2c7756 (diff) | |
| download | ComputeLibrary-51403b5b8ae92782e2ed467cae4c0783faa5a22b.tar.gz | |
COMPMID-2336: Fix argument type for epsilon in CLL2NormalizeLayer.
Change-Id: Iee3ef6d022f497df0e2e6227e06462762042b394
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1367
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 'src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp index 00af590104..288e1e8e61 100644 --- a/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp +++ b/src/core/CL/kernels/CLL2NormalizeLayerKernel.cpp @@ -100,11 +100,11 @@ void CLL2NormalizeLayerKernel::configure(const ICLTensor *input, const ICLTensor ARM_COMPUTE_ERROR_ON_NULLPTR(input, sum, output); ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(input->info(), sum->info(), output->info(), axis, epsilon)); - _input = input; - _sum = sum; - _output = output; - _actual_axis = wrap_around(axis, max_input_tensor_dim); - _epsilon = epsilon; + _input = input; + _sum = sum; + _output = output; + _actual_axis = wrap_around(axis, max_input_tensor_dim); + _epsilon = epsilon; const unsigned int num_elems_processed_per_iteration = 16; @@ -138,11 +138,11 @@ void CLL2NormalizeLayerKernel::configure(const ICLTensor *input, const ICLTensor // Set epsilon argument if(input->info()->data_type() == DataType::F32) { - _kernel.setArg<cl_uint>(idx, _epsilon); + _kernel.setArg<cl_float>(idx, _epsilon); } else { - _kernel.setArg<cl_ushort>(idx, _epsilon); + _kernel.setArg<cl_half>(idx, _epsilon); } // Configure kernel window |