提交 7de4770a 编写于 作者: H hjchen2

Set winograd usage conditions to use im2col+gemm for small input

上级 c0f7ecb4
......@@ -56,10 +56,9 @@ void InitBaseConvKernel(ConvParam<CPU> *param) {
param->Strides()[0] == param->Strides()[1] &&
param->Dilations()[0] == param->Dilations()[1] &&
param->Strides()[0] == 1 && param->Dilations()[0] == 1
#if 0
&& param->Output()->dims()[1] >= 16 &&
param->Input()->dims()[1] >= 16 &&
param->Input()->dims()[2] <= 140 */ /* refered from ncnn */
#if 1
&& param->Input()->dims()[1] >= 4
&& param->Input()->dims()[2] >= 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.
先完成此消息的编辑!
想要评论请 注册