未验证 提交 b2db5f49 编写于 作者: T TianXiaogang 提交者: GitHub

fix: fix fc batch bug (#2244)

* fix: fix fc batch bug

* fix:fix fc bug;test=develop
上级 362275ed
......@@ -88,8 +88,7 @@ void FcCompute<PRECISION(kFloat), PRECISION(kFloat)>::Run() {
auto i_data = param.input->data<float>();
auto o_data = param.output->mutable_data<float>();
auto w_data =
flag_trans_weights_ ? weights_.data<float>() : param.w->data<float>();
auto w_data = flag_gemm_ ? param.w->data<float>() : weights_.data<float>();
const float* b_data = param.bias ? param.bias->data<float>() : nullptr;
if (flag_trans_bias_) {
b_data = bias_.data<float>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册