diff --git a/mindspore/lite/tools/converter/graphdef_transform.cc b/mindspore/lite/tools/converter/graphdef_transform.cc index 312bbfdb9c..965cf7473c 100644 --- a/mindspore/lite/tools/converter/graphdef_transform.cc +++ b/mindspore/lite/tools/converter/graphdef_transform.cc @@ -97,17 +97,6 @@ void GraphDefTransform::CreateQuantizer(const converter::Flags *flags) { int GraphDefTransform::Transform(const converter::Flags &ctx) { STATUS status; - { - Optimizer fusionOptimizer; - fusionOptimizer.AddPass(new (std::nothrow) FormatTransPermuteFusionPass()); - fusionOptimizer.AddPass(new (std::nothrow) IsolatedNodeRemovePass()); - status = fusionOptimizer.Run(graphDefT); - if (status != RET_OK && status != RET_NO_CHANGE) { - MS_LOG(ERROR) << "Run fusionOptimizer graphPasses Failed"; - return status; - } - } - // weight format trans if (ctx.formatTrans) { Optimizer weightFormatOptimizer; @@ -195,6 +184,16 @@ int GraphDefTransform::Transform(const converter::Flags &ctx) { return status; } } + { + Optimizer fusionOptimizer; + fusionOptimizer.AddPass(new (std::nothrow) FormatTransPermuteFusionPass()); + fusionOptimizer.AddPass(new (std::nothrow) IsolatedNodeRemovePass()); + status = fusionOptimizer.Run(graphDefT); + if (status != RET_OK && status != RET_NO_CHANGE) { + MS_LOG(ERROR) << "Run fusionOptimizer graphPasses Failed"; + return status; + } + } // do quantization if (fbQuantizer != nullptr) { diff --git a/mindspore/lite/tools/converter/legacy_optimizer/fusion/format_trans_transpose_fusion_pass.cc b/mindspore/lite/tools/converter/legacy_optimizer/fusion/format_trans_transpose_fusion_pass.cc index 9a81e60dce..a40b03e38b 100644 --- a/mindspore/lite/tools/converter/legacy_optimizer/fusion/format_trans_transpose_fusion_pass.cc +++ b/mindspore/lite/tools/converter/legacy_optimizer/fusion/format_trans_transpose_fusion_pass.cc @@ -58,7 +58,7 @@ STATUS FormatTransPermuteFusionPass::DefinePattern() { formatTransOp->types = {PrimitiveType_Nchw2Nhwc, PrimitiveType_Nhwc2Nchw}; auto transposeOp = std::make_shared(); transposeOp->id = kPermuteOp; - transposeOp->types = {PrimitiveType_Permute}; + transposeOp->types = {PrimitiveType_Transpose}; formatTransOp->left = transposeOp; std::unique_ptr transposeFormatTransFusionPattern(