From 7e56655bc286ad51ffccb32f3bad36a5ac24a9be Mon Sep 17 00:00:00 2001 From: zhengjun10 Date: Fri, 9 Oct 2020 09:43:36 +0800 Subject: [PATCH] weight format transform modify --- mindspore/lite/tools/converter/anf_transform.cc | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/mindspore/lite/tools/converter/anf_transform.cc b/mindspore/lite/tools/converter/anf_transform.cc index d5f90a00e5..feb758278d 100644 --- a/mindspore/lite/tools/converter/anf_transform.cc +++ b/mindspore/lite/tools/converter/anf_transform.cc @@ -63,15 +63,15 @@ FuncGraphPtr AnfTransform::Transform(const FuncGraphPtr &old_graph, const conver true, "conv_tuple_relu", schema::PrimitiveType_Activation, schema::ActivationType_RELU)); pm->AddPass(std::make_shared( true, "conv_tuple_relu6", schema::PrimitiveType_Activation, schema::ActivationType_RELU6)); - auto weight_format_hardcode_pass = std::make_shared(); - weight_format_hardcode_pass->SetFmkType(config->fmk); - weight_format_hardcode_pass->SetQuantType(config->quantType); - graph_pm->AddPass(weight_format_hardcode_pass); - auto weight_format_transform_pass = std::make_shared(); - weight_format_transform_pass->SetFmkType(config->fmk); - weight_format_transform_pass->SetQuantType(config->quantType); - graph_pm->AddPass(weight_format_transform_pass); } + auto weight_format_hardcode_pass = std::make_shared(); + weight_format_hardcode_pass->SetFmkType(config->fmk); + weight_format_hardcode_pass->SetQuantType(config->quantType); + graph_pm->AddPass(weight_format_hardcode_pass); + auto weight_format_transform_pass = std::make_shared(); + weight_format_transform_pass->SetFmkType(config->fmk); + weight_format_transform_pass->SetQuantType(config->quantType); + graph_pm->AddPass(weight_format_transform_pass); if (config->fmk == lite::converter::FmkType_MS) { auto remove_unused_cast_pass = std::make_shared();