diff --git a/ge/graph/optimize/mem_rw_conflict_optimize.cc b/ge/graph/optimize/mem_rw_conflict_optimize.cc
index 30fcabef..7e7ab908 100644
--- a/ge/graph/optimize/mem_rw_conflict_optimize.cc
+++ b/ge/graph/optimize/mem_rw_conflict_optimize.cc
@@ -743,10 +743,8 @@ Status GraphOptimize::HandleMemoryRWConflict(ComputeGraphPtr &compute_graph) {
       continue;
     }
     // ignore data / netoutput of subgraph
-    if (AttrUtils::HasAttr(node->GetOpDesc(), ATTR_NAME_PARENT_NODE_INDEX)) {
-      if (node->GetType() == DATA || node->GetType() == NETOUTPUT) {
-        continue;
-      }
+    if (IsSubgraphInputNode(node) || IsSubgraphOutputNode(node)) {
+      continue;
     }
 
     bool identity_reserved = false;