!10246 [lite] adjust tf and mindir nit suport op's print

From: @xu_anyue
Reviewed-by: @hangangqiang
Signed-off-by: @hangangqiang

mindspore/lite/tools/converter/parser/tf/tf_model_parser.cc

@@ -306,6 +306,7 @@ STATUS TFModelParser::ConvertGraphInputsAndConsts(
 FuncGraphPtr paserTfFuction() { return nullptr; }
 FuncGraphPtr TFModelParser::Parse(const std::string &modelFile, const std::string &weightFile,
                                   const QuantType &quantType) {
+  NoSupportOp::GetInstance()->SetFmkType("TF");
   auto status = ValidateFileStr(modelFile, ".pb");
   if (status != RET_OK) {
     MS_LOG(ERROR) << "INPUT ILLEGAL: modelFile must be *.pb";
@@ -321,7 +322,7 @@ FuncGraphPtr TFModelParser::Parse(const std::string &modelFile, const std::strin
   status = ReadProtoFromBinaryFile((const char *)modelFile.c_str(), tf_root_graph_.get());
   if (status != RET_OK) {
     MS_LOG(ERROR) << "Open modelFile for TF converter failed!";
-    ReturnCode::GetSingleReturnCode()->UpdateReturnCode(RET_ERROR);
+    ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
     return nullptr;
   }
   anf_root_graph_ = std::make_shared<FuncGraph>();
@@ -346,13 +347,13 @@ FuncGraphPtr TFModelParser::Parse(const std::string &modelFile, const std::strin
   for (int i = 0; i < tf_root_graph_->node_size(); i++) {
     auto &node_def = tf_root_graph_->node(i);
     status = ConvertOps(node_def, tf_root_graph_nodes_, anf_root_graph_, &anf_root_node_map_);
+    ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
     if (status != RET_OK) {
       success_flag = false;
     }
   }
   if (!success_flag) {
     MS_LOG(ERROR) << "Convert ops failed.";
-    ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
     return nullptr;
   }
   status = ConvertRootGraphOutputs();
@@ -376,6 +377,7 @@ STATUS TFModelParser::ConvertSubgraph() {
   auto subgraph_size = graph_def_liarary.function_size();
   std::map<CNodePtr, FuncGraphPtr> while_cond_map;
   std::map<CNodePtr, FuncGraphPtr> while_body_map;
+  bool success_flag = true;
   for (int i = 0; i < subgraph_size; i++) {
     auto &tf_sub_fuction = graph_def_liarary.function(i);
     auto &tf_sub_signature = tf_sub_fuction.signature();
@@ -421,12 +423,16 @@ STATUS TFModelParser::ConvertSubgraph() {
     for (int j = 0; j < tf_sub_fuction.node_def_size(); j++) {
       auto &node_def = tf_sub_fuction.node_def(j);
       status = ConvertOps(node_def, tf_sub_node_map, sub_func_graph, &anf_sub_node_map);
+      ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
       if (status != RET_OK) {
         MS_LOG(ERROR) << "Convert subgraph ops failed.";
-        ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
+        success_flag = false;
-        return RET_ERROR;
       }
     }
+    if (!success_flag) {
+      MS_LOG(ERROR) << "Convert subgraph is failed.";
+      continue;
+    }
 
     // convert subgraph outputs
     std::vector<AnfNodePtr> sub_output_nodes;
@@ -483,6 +489,10 @@ STATUS TFModelParser::ConvertSubgraph() {
 
     MS_LOG(INFO) << "parse subgraph end:" << sub_graph_name;
   }
+  if (!success_flag) {
+    MS_LOG(ERROR) << "Convert subgraph is failed.";
+    return RET_ERROR;
+  }
   auto status = WhileNodePostProcess(while_cond_map, while_body_map);
   if (status != RET_OK) {
     MS_LOG(ERROR) << "while node post process failed";
@@ -593,7 +603,6 @@ STATUS TFModelParser::ConvertOps(const tensorflow::NodeDef &node_def,
                                  std::unordered_map<std::string, AnfNodePtr> *anf_node_map) {
   MS_ASSERT(node_def != nullptr);
   MS_ASSERT(func_graph_ptr != nullptr);
-  NoSupportOp::GetInstance()->SetFmkType("TF");
   STATUS status = RET_OK;
   const auto &op_type = node_def.op();
   if (op_type == "Placeholder" || op_type == "Const" || op_type == "Identity" || op_type == "StopGradient") {
@@ -645,8 +654,6 @@ STATUS TFModelParser::ConvertOps(const tensorflow::NodeDef &node_def,
   status = ConvertOutputTensor(node_def, anf_node, anf_node_map, func_graph_ptr, output_size);
   if (status != RET_OK) {
     MS_LOG(ERROR) << "Convert output tensors for " << anf_node->fullname_with_scope() << " failed.";
-    ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
-    return RET_ERROR;
   }
   return status;
 }

mindspore/lite/tools/optimizer/graph/mindir_adjust_pass.cc

@@ -19,6 +19,7 @@
 #include <memory>
 
 #include "src/ops/primitive_c.h"
+#include "tools/converter/converter_context.h"
 #include "tools/converter/quantizer/quant_cast.h"
 #include "src/common/log_adapter.h"
 #include "src/tensor.h"
@@ -111,7 +112,8 @@ int MindirAdjustPass::PrimitiveConvert(std::shared_ptr<AnfNode> anf_node) {
     auto primitive_c = PrimitiveC::Create(*primitive, inputs, quant_type_);
     if (primitive_c == nullptr) {
       MS_LOG(ERROR) << "fail to create a primitive_c: " << cnode->fullname_with_scope();
-      return lite::RET_ERROR;
+      lite::NoSupportOp::GetInstance()->InsertOp(primitive->name());
+      return lite::RET_NOT_FIND_OP;
     }
     value_node->set_value(primitive_c);
   } else {
@@ -131,6 +133,7 @@ bool MindirAdjustPass::Run(const FuncGraphPtr &graph) {
   MS_ASSERT(graph != nullptr);
   auto node_list = TopoSort(graph->get_return());
   int status = lite::RET_OK;
+  bool success_flag = true;
   for (auto &node : node_list) {
     if (utils::isa<ParameterPtr>(node)) {
       status = ParameterNodeConvert(node);
@@ -138,9 +141,14 @@ bool MindirAdjustPass::Run(const FuncGraphPtr &graph) {
       status = PrimitiveConvert(node);
     }
     if (status != lite::RET_OK && status != lite::RET_NO_CHANGE) {
-      return false;
+      lite::ReturnCode::GetSingleReturnCode()->UpdateReturnCode(status);
+      success_flag = false;
     }
   }
+  if (!success_flag) {
+    MS_LOG(ERROR) << "Adjust mindir failed.";
+    return false;
+  }
   return true;
 }
 }  // namespace opt