Merge branch 'opted_with_output'

# Conflicts:
#	examples/onnx_output_model/model_opted.onnx
#	examples/optimizer_c_example/predicate_example.cpp

Author: yangxinyu7427

Diff for: onnxoptimizer/query_c_api/predicate_push_down.cpp

@@ -100,17 +100,24 @@ void merge_single_model_with_predicate(std::string& onnx_model_path, std::string
   *onnx_model.mutable_graph()->add_node()=reshape_node;
   *onnx_model.mutable_graph()->add_input()=input;
   *onnx_model.mutable_graph()->add_node()=node;
+  std::regex pattern_output_label("_output_label$");
   std::vector<onnx::ValueInfoProto> output_list;
+  std::vector<onnx::ValueInfoProto> save_output_list;
   for(int i=0;i<onnx_model.graph().output_size();i++){
-    if(!std::regex_search(onnx_model.graph().output(i).name(), pattern)){
+    if(!std::regex_search(onnx_model.graph().output(i).name(), pattern))
       output_list.push_back(onnx_model.graph().output(i));
-    }
+    else if(std::regex_search(onnx_model.graph().output(i).name(), pattern_output_label))
+      save_output_list.push_back(onnx_model.graph().output(i));
   }
   onnx_model.mutable_graph()->clear_output();
+  *onnx_model.mutable_graph()->add_output()=output;
   for(int i=0;i<output_list.size();i++){
     *onnx_model.mutable_graph()->add_output()=output_list.at(i);
   }
-  *onnx_model.mutable_graph()->add_output()=output;
+  for(int i=0;i<save_output_list.size();i++){
+    *onnx_model.mutable_graph()->add_output()=save_output_list.at(i);
+  }
+
 
   onnx::checker::check_model(onnx_model);
   saveModel(&onnx_model,onnx_model_path);
@@ -141,7 +148,7 @@ void merge_double_models_with_predicate(std::string& onnx_model_path,std::string
   onnx::ValueInfoProto output_l;
   onnx::ValueInfoProto output_r;
 
-  for(int i=0;i<onnx_model.graph().output_size();i++){
+  for(int i=onnx_model.graph().output_size()-1;i>=0;i--){
       if(std::regex_search(onnx_model.graph().output(i).name(), pattern_l)&&
         !std::regex_search(onnx_model.graph().output(i).name(), pattern_end)&&
         !std::regex_search(onnx_model.graph().output(i).name(), pattern_ends)){
@@ -168,18 +175,25 @@ void merge_double_models_with_predicate(std::string& onnx_model_path,std::string
 
   // change onnx_model
   *onnx_model.mutable_graph()->add_node()=node;
+  std::regex pattern_output_label("_output_label$");
   std::vector<onnx::ValueInfoProto> output_list;
+  std::vector<onnx::ValueInfoProto> save_output_list;
   for(int i=0;i<onnx_model.graph().output_size();i++){
       if(!std::regex_search(onnx_model.graph().output(i).name(), pattern_l) &&
           !std::regex_search(onnx_model.graph().output(i).name(), pattern_r)){
         output_list.push_back(onnx_model.graph().output(i));
-      }
+      } else if(std::regex_search(onnx_model.graph().output(i).name(), pattern_output_label))
+        save_output_list.push_back(onnx_model.graph().output(i));
   }
   onnx_model.mutable_graph()->clear_output();
+  *onnx_model.mutable_graph()->add_output()=output;
   for(int i=0;i<output_list.size();i++){
       *onnx_model.mutable_graph()->add_output()=output_list.at(i);
   }
-  *onnx_model.mutable_graph()->add_output()=output;
+  for(int i=0;i<save_output_list.size();i++){
+      *onnx_model.mutable_graph()->add_output()=save_output_list.at(i);
+  }
+
 
   onnx::checker::check_model(onnx_model);
   saveModel(&onnx_model,onnx_model_path);