Fix the wrong gemm_compatible()

JDAI-CV · May 29, 2019 · 5eda6d2 · 5eda6d2
1 parent b68345b
commit 5eda6d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -87,7 +87,7 @@ bool BinConv::direct_conv_compatible() const {
 
 bool BinConv::gemm_compatible() const {
 #ifdef __aarch64__
-    return weight_mat->h * weight_mat->n * weight_mat->c % 2 == 0;
+    return weight_mat->h * weight_mat->w * weight_mat->c % 2 == 0;
 #else
     return false;
 #endif