diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2017-10-03 13:17:02 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | ece307bd365ce0be0be7168fdafa95671480dbbb (patch) | |
| tree | b02a3906e787a96e3ffb6146b36e7fcf34243de5 /src/runtime/NEON/functions/NEConvolutionLayer.cpp | |
| parent | 6f669f039fb74675b858bc3703295609a6a3e122 (diff) | |
| download | ComputeLibrary-ece307bd365ce0be0be7168fdafa95671480dbbb.tar.gz | |
COMPMID-556 - Fix ConvolutionLayer test
Change-Id: I8035cfd2b7fd62bd9f79f2a7d4b59f3e2bbe9163
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/89988
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEConvolutionLayer.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEConvolutionLayer.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp index 40862fc2d8..f34f497436 100644 --- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp @@ -190,8 +190,9 @@ void NEConvolutionLayer::configure(const ITensor *input, const ITensor *weights, { if(_are_weights_reshaped) { - mat_weights_cols = weights_info.num_kernels(); - mat_weights_rows = weights->info()->dimension(0) / 4 + (_has_bias ? 1 : 0); + const unsigned int transpose_width = 16 / input->info()->element_size(); + mat_weights_cols = weights_info.num_kernels(); + mat_weights_rows = weights->info()->dimension(0) / transpose_width + (_has_bias ? 1 : 0); } else { |