提交 60889629 编写于 作者: S smilejames 提交者: GitHub

Merge pull request #863 from smilejames/develop

separate C = A * B + bias and C = A * B + C in Gemm
...@@ -674,11 +674,19 @@ void InnerKernelWithBias(int mc, int nc, float alpha, const float *a, ...@@ -674,11 +674,19 @@ void InnerKernelWithBias(int mc, int nc, float alpha, const float *a,
return; return;
} }
if (beta == 1 && !relu) { if (beta == 1 && !relu) {
WriteWithAddV1(mc, nc, c, C, ldc, bias); if (bias == nullptr) {
WriteWithAdd(mc, nc, c, C, ldc);
} else {
WriteWithAddV1(mc, nc, c, C, ldc, bias);
}
return; return;
} }
if (beta == 1 && relu) { if (beta == 1 && relu) {
WriteWithAddReluV1(mc, nc, c, C, ldc, bias); if (bias == nullptr) {
WriteWithAddRelu(mc, nc, c, C, ldc);
} else {
WriteWithAddReluV1(mc, nc, c, C, ldc, bias);
}
return; return;
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册