diff --git a/dnn/src/fallback/conv_bias/winograd/winograd.h b/dnn/src/fallback/conv_bias/winograd/winograd.h index eb1796ee..fda01133 100644 --- a/dnn/src/fallback/conv_bias/winograd/winograd.h +++ b/dnn/src/fallback/conv_bias/winograd/winograd.h @@ -214,6 +214,11 @@ public: SmallVector deduce_preprocessed_filter_layout( const NCBKernSizeParam& param, fallback::MatrixMulImpl::AlgoBase*) { + if (param.filter_meta.format != param::ConvBias::Format::NCHW && + param.filter_meta.format != param::ConvBias::Format::NCHW88 && + param.filter_meta.format != param::ConvBias::Format::NCHW44) { + return {}; + } size_t OC = param.filter_meta.ocpg; size_t IC = param.filter_meta.icpg; size_t GROUP = param.filter_meta.group;