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

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

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