diff options
Diffstat (limited to 'src/cpu/kernels/assembly')
-rw-r--r-- | src/cpu/kernels/assembly/arm_gemm.hpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/cpu/kernels/assembly/arm_gemm.hpp b/src/cpu/kernels/assembly/arm_gemm.hpp index 9920b863d9..247cb1d470 100644 --- a/src/cpu/kernels/assembly/arm_gemm.hpp +++ b/src/cpu/kernels/assembly/arm_gemm.hpp @@ -143,12 +143,12 @@ struct GemmArgs { public: const CPUInfo *_ci; - unsigned int _Msize; - unsigned int _Nsize; - unsigned int _Ksize; + unsigned int _Msize; // num of tiles + unsigned int _Nsize; // output channels + unsigned int _Ksize; // input channels unsigned int _Ksections; unsigned int _nbatches; - unsigned int _nmulti; + unsigned int _nmulti; // n_gemms to be performed bool _indirect_input; Activation _act; int _maxthreads; |