OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

[MSLITE] onnx deconv convertor bug

This commit is contained in:
ling 2020-09-21 11:36:11 +08:00
parent 8415ab53f6
commit 6e62de56d1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/tools/converter/legacy_optimizer/graph/weight_format_hardcode_pass.cc
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;