Browse Source

fix(dnn): fix nchw88 winograd weight preprocess

GitOrigin-RevId: 03c0c52aba
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
651920c7ad
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/opr/impl/dnn/convolution.cpp

+ 1
- 1
src/opr/impl/dnn/convolution.cpp View File

@@ -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;
}


Loading…
Cancel
Save