From b68345bd6b2987de00b5c046f320f2871fcde934 Mon Sep 17 00:00:00 2001 From: daquexian Date: Wed, 29 May 2019 16:58:50 +0800 Subject: [PATCH] Add the missing require_align=False for trans_weight_mat --- dabnn/layers/BinConv.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dabnn/layers/BinConv.cpp b/dabnn/layers/BinConv.cpp index c8479d2..1ffe8ab 100644 --- a/dabnn/layers/BinConv.cpp +++ b/dabnn/layers/BinConv.cpp @@ -46,7 +46,7 @@ BinConv::BinConv(NetCP net, const std::string &name, css input, css weight, const int k = weight_mat->h * weight_mat->w * weight_mat->c; transposed_weight_mat = std::make_shared(weight_mat->n, weight_mat->h, weight_mat->w, - weight_mat->elem_c, DataType::Bit); + weight_mat->elem_c, DataType::Bit, false); auto *trans_data_ptr = static_cast(transposed_weight_mat->data); auto *data_ptr = static_cast(weight_mat->data);