diff --git a/mindspore/lite/tools/converter/legacy_optimizer/graph/switch_pass.cc b/mindspore/lite/tools/converter/legacy_optimizer/graph/switch_pass.cc
index 773f1c4b7d..5feb076f63 100644
--- a/mindspore/lite/tools/converter/legacy_optimizer/graph/switch_pass.cc
+++ b/mindspore/lite/tools/converter/legacy_optimizer/graph/switch_pass.cc
@@ -209,6 +209,7 @@ STATUS SingleSwitchPass::InsertMerge() {
       merge_node->inputIndex.push_back(old_merge_input[i]);
     } else {
       auto tensor = NewTensor(in_tensor);
+      tensor->nodeType = schema::NodeType_CNode;
       graph_->allTensors.push_back(std::move(tensor));
       merge_node->inputIndex.push_back(graph_->allTensors.size() - 1);
     }
diff --git a/mindspore/lite/tools/converter/parser/tf/tf_node_parser.cc b/mindspore/lite/tools/converter/parser/tf/tf_node_parser.cc
index 54ab046b9b..2d6f726c3c 100644
--- a/mindspore/lite/tools/converter/parser/tf/tf_node_parser.cc
+++ b/mindspore/lite/tools/converter/parser/tf/tf_node_parser.cc
@@ -37,6 +37,14 @@ const NodeDef *TFNodeParser::GetConstInputNode(const std::map<string, const tens
     return nullptr;
   }
   auto node = tf_node_map.at(flatten_input_name);
+  // node is "Identity Const"
+  if (node->op() == "Identity") {
+    flatten_input_name = TensorFlowUtils::GetFlattenNodeName(node->input(0));
+    if (tf_node_map.find(flatten_input_name) == tf_node_map.end()) {
+      return nullptr;
+    }
+    node = tf_node_map.at(flatten_input_name);
+  }
   if (node->op() != "Const") {
     MS_LOG(ERROR) << "Attr node is not Const";
     return nullptr;