diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2017-08-15 11:45:22 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | edfa9f463bed084f8b0953557202b2a1e56da817 (patch) | |
| tree | 5d1e92926d112fde05dcbc61324d96f73f692390 /src/runtime/CL/functions/CLConvolutionLayer.cpp | |
| parent | dc460f13ee65e27b2a428e44c2d80afb1f516a99 (diff) | |
| download | ComputeLibrary-edfa9f463bed084f8b0953557202b2a1e56da817.tar.gz | |
COMPMID-477 - Optimized batched case in CLConvolutionLayer
Change-Id: I4ef18f49f1da0cb816aaa0762466b940792c15ed
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/84162
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLConvolutionLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLConvolutionLayer.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLConvolutionLayer.cpp b/src/runtime/CL/functions/CLConvolutionLayer.cpp index b1b83985d0..0bbec94e78 100644 --- a/src/runtime/CL/functions/CLConvolutionLayer.cpp +++ b/src/runtime/CL/functions/CLConvolutionLayer.cpp @@ -197,9 +197,12 @@ void CLConvolutionLayer::configure(const ICLTensor *input, const ICLTensor *weig // Configure kernels _input_im2col_kernel.configure(input, &_input_im2col_reshaped, Size2D(kernel_width, kernel_height), conv_info, _has_bias); + + // Configure matrix multiply if(_is_fully_connected_convolution) { - _mm_kernel.configure(&_input_im2col_reshaped, weights, &_gemm_output, 1.0f); + // The matrix A and Matrix B have not been reshaped + _mm_kernel.configure(&_input_im2col_reshaped, weights, &_gemm_output, 1.0f, false); } else { |