Fix bug: output_mat is not inited in bgemm

JDAI-CV · May 29, 2019 · 5af2ec3 · 5af2ec3
1 parent 8559844
commit 5af2ec3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp
@@ -99,6 +99,7 @@ void BinConv::forward_impl() const {
             pad(*input_mat, pad_h, pad_w, *padded_mat);
             bconv_3x3(*padded_mat, *weight_mat, *output_mat, stride_h);
         } else if (gemm_compatible()) {
+            output_mat->fill<float>(0.f);
             bnn::im2col(*input_mat, weight_mat->h, weight_mat->w, pad_h, pad_w,
                         stride_h, stride_w, 1, 1, *col_mat);
             const int m = weight_mat->n;