m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6625 [MSLITE] onnx deconv convertor bug

Browse Source 
Merge pull request !6625 from ling/bug
pull/6625/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent ad706b5938 6e62de56d1
commit da66ba8b61
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
mindspore/lite/tools/converter/legacy_optimizer/graph/weight_format_hardcode_pass.cc
Unescape View File

@ -118,7 +118,7 @@ STATUS WeightFormatHardCodePass::HardCodeONNX(const std::unique_ptr<CNodeT> &nod
} else if (opType == PrimitiveType_DepthwiseConv2D) {
weightTensor->format = schema::Format::Format_CHWK;
} else if (opType == PrimitiveType_DeConv2D) {
weightTensor->format = schema::Format::Format_CKHW;
weightTensor->format = schema::Format::Format_KCHW;
} else {
MS_LOG(ERROR) << "Unsupported opType: " << EnumNamePrimitiveType(opType) << ", node: " << node->name;
return RET_ERROR;
@ -133,7 +133,7 @@ STATUS WeightFormatHardCodePass::HardCodeONNX(const std::unique_ptr<CNodeT> &nod
if (opType == PrimitiveType_Conv2D || opType == PrimitiveType_DepthwiseConv2D) {
weightTensor->format = schema::Format::Format_KCHW;
} else if (opType == PrimitiveType_DeConv2D) {
weightTensor->format = schema::Format::Format_CKHW;
weightTensor->format = schema::Format::Format_KCHW;
} else {
MS_LOG(ERROR) << "Unsupported opType: " << EnumNamePrimitiveType(opType) << ", node: " << node->name;
return RET_ERROR;

Write Preview
Loading…
Cancel
Save