Browse Source

!1851 add atc_params: check_report for ConvertModelToJson

From: @lichun30
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
tags/v1.5.1
mindspore-ci-bot Gitee 3 years ago
parent
commit
1af22f128c
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/offline/main.cc

+ 1
- 0
ge/offline/main.cc View File

@@ -863,6 +863,7 @@ domi::Status GenerateInfershapeJson() {
ge::Graph graph;
std::map<string, string> atc_params;
atc_params.insert(std::pair<string, string>("input_format", FLAGS_input_format));
atc_params.insert(std::pair<string, string>("check_report", FLAGS_check_report));
ret = ParseGraph(graph, atc_params, FLAGS_om.c_str(), FLAGS_weight.c_str(), (domi::FrameworkType) FLAGS_framework,
"", FLAGS_target.c_str(), (ge::RunMode) FLAGS_mode, false);
if (ret != ge::SUCCESS) {


Loading…
Cancel
Save