diff options
| author | Anton Lokhmotov <psyhtest@users.noreply.github.com> | 2017-11-08 09:34:19 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | af6204c331eed7894ec4c5fd4e98ec22b6dac676 (patch) | |
| tree | 7c952fbcf3ec11ff4c9d49061be12742e9cd979d /src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp | |
| parent | 3a873a578e80481a55ce3b885078948ae79468eb (diff) | |
| download | ComputeLibrary-af6204c331eed7894ec4c5fd4e98ec22b6dac676.tar.gz | |
COMPMID-661: Add avgpool-uint8 support. Optimize avgpool-fp32 for Bifrost. (#13)
Change-Id: I32ba6afbac6694ffa053dd16f03a1b3d14627a19
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/94857
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp')
| -rw-r--r-- | src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp index 53e46390c1..5f109f76af 100644 --- a/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp +++ b/src/core/CL/kernels/CLDirectConvolutionLayerKernel.cpp @@ -84,7 +84,12 @@ void CLDirectConvolutionLayerKernel::configure(const ICLTensor *input, const ICL output_shape.set(2, weights->info()->dimension(3)); // Output auto inizialitation if not yet initialized - auto_init_if_empty(*output->info(), output_shape, 1, input->info()->data_type(), input->info()->fixed_point_position()); + auto_init_if_empty(*output->info(), + output_shape, + 1, + input->info()->data_type(), + input->info()->fixed_point_position(), + input->info()->quantization_info()); ARM_COMPUTE_ERROR_ON_MISMATCHING_DIMENSIONS(output->info()->tensor_shape(), output_shape); ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(input, output); @@ -176,7 +181,7 @@ void CLDirectConvolutionLayerKernel::configure(const ICLTensor *input, const ICL else { bool is_quantized_fixed_point = is_data_type_fixed_point(data_type); - bool is_quantized_asymm = is_data_type_quantized_assymetric(data_type); + bool is_quantized_asymm = is_data_type_quantized_asymmetric(data_type); DataType promoted_type = (is_quantized_fixed_point) ? get_promoted_data_type(data_type) : data_type; build_options.add_option_if(is_quantized_asymm, std::string("-DKERNEL_SIZE=" + support::cpp11::to_string(kernel_size))); @@ -220,7 +225,7 @@ void CLDirectConvolutionLayerKernel::configure(const ICLTensor *input, const ICL } // Set static kernel arguments - if(is_data_type_quantized_assymetric(data_type)) + if(is_data_type_quantized_asymmetric(data_type)) { int output_multiplier = 0; int output_shift = 0; |