提交 651920c7 编写于 作者: M Megvii Engine Team

fix(dnn): fix nchw88 winograd weight preprocess

GitOrigin-RevId: 03c0c52abaa870594e17472814a81d6286cdabf8
上级 adf5620e
......@@ -794,7 +794,7 @@ void AlgoChooser<megdnn::ConvBias>::ExeContext::
param.opr_param.format =
megdnn::ConvBias::Param::Format::NCHW44_WINOGRAD;
} else if (param.opr_param.format ==
megdnn::ConvBias::Param::Format::NCHW) {
megdnn::ConvBias::Param::Format::NCHW88) {
param.opr_param.format =
megdnn::ConvBias::Param::Format::NCHW88_WINOGRAD;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册