commit bf8b01dfbfdca124673ade33c5eac8f3748d7abd
author Kaizen <kaizen@arm.com> Thu Oct 12 14:26:51 2017 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Thu Oct 12 15:42:58 2017 +0100
tree 9504f134ca292b988e7065b22229e9bbe0493d00
parent 8938bd3f40ea62ff56d6ed4e2db0a8aee34dd64a

arm_compute v17.10

Change-Id: If1489af40eccd0219ede8946577afbf04db31b29

src/runtime/NEON/functions/NEConvolutionLayer.cpp

diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp
index 40862fc..f34f497 100644
--- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp
+++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp
@@ -190,8 +190,9 @@
     {
         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
         {