Browse Source

add atc_params: check_report for ConvertModelToJson

tags/v1.5.1
lichun 4 years ago
parent
commit
17a37ca750
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/offline/main.cc

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

@@ -847,6 +847,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