Browse Source

!6625 [MSLITE] onnx deconv convertor bug

Merge pull request !6625 from ling/bug
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
da66ba8b61
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/lite/tools/converter/legacy_optimizer/graph/weight_format_hardcode_pass.cc

+ 2
- 2
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;


Loading…
Cancel
Save