提交 5245344d 编写于 作者: H hjchen2

Refine winograd conditions for googlenet

上级 320528ad
......@@ -57,8 +57,8 @@ void InitBaseConvKernel(ConvParam<CPU> *param) {
param->Dilations()[0] == param->Dilations()[1] &&
param->Strides()[0] == 1 && param->Dilations()[0] == 1
#if 1
&& param->Input()->dims()[1] >= 4
&& param->Input()->dims()[2] >= 16
&& (param->Input()->dims()[1] >= 4 ||
param->Output()->dims()[1] >= 16)
#endif
) {
param->ExecMode() = ConvParam<CPU>::EXEC_WINOGRAD3X3_FLOAT;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册