weight->c==1 -> input_mat->elem_c==64

JDAI-CV · Aug 22, 2019 · 22fe7c4 · 22fe7c4
1 parent 28ec3a5
commit 22fe7c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -130,11 +130,11 @@ bool BinConv::gemm_compatible() const {
 #ifdef __aarch64__
     return true;
 #else
-    // If weight_mat->c == 1 (weight_mat has 64 channels), we use bconv_64
+    // If input_mat->elem_c == 1 (weight_mat has 64 channels), we use bconv_64
     // in aarch64 for the fastest speed, however, bconv_64 is not implemented
     // in armv7
     // TODO: Implement bconv_64 for armv7
-    return weight_mat->c != 1;
+    return input_mat->elem_c != 64;
 #endif
 #else
     return false;