From 651920c7adb5880008f56f508b77ff125ad64561 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 7 Sep 2020 12:22:21 +0800 Subject: [PATCH] fix(dnn): fix nchw88 winograd weight preprocess GitOrigin-RevId: 03c0c52abaa870594e17472814a81d6286cdabf8 --- src/opr/impl/dnn/convolution.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/opr/impl/dnn/convolution.cpp b/src/opr/impl/dnn/convolution.cpp index 1180695d..d00d972f 100644 --- a/src/opr/impl/dnn/convolution.cpp +++ b/src/opr/impl/dnn/convolution.cpp @@ -794,7 +794,7 @@ void AlgoChooser::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; }