Fix some missing cases for binconv method

JDAI-CV · Aug 22, 2019 · 28ec3a5 · 28ec3a5
1 parent c472484
commit 28ec3a5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -83,11 +83,13 @@ BinConv::Method BinConv::method() const {
             return Method::DIRECT_CONV;
         } else if (gemm_compatible()) {
             return Method::BGEMM;
-        } else {
+        } else if (input_mat->elem_c == 64) {
             return Method::BCONV_NAIVE;
+        } else {
+            return Method::BGEMM_NAIVE;
         }
     } else {
-        if (weight_mat->c == 1) {
+        if (input_mat->elem_c == 64) {
             return Method::BCONV_NAIVE;
         } else {
             return Method::BGEMM_NAIVE;