|
|
@@ -63,7 +63,7 @@ void TransOpWithoutReshapeFusionPass::SetRemainNode( |
|
|
|
continue; |
|
|
|
} |
|
|
|
GELOGI("SetRemainNode node is %s", op_desc->GetName().c_str()); |
|
|
|
GE_IF_BOOL_EXEC(!op_desc->SetExtAttr(kRemainNode, true), GELOGE(INTERNAL_ERROR, "set ext attr failed"); return ); |
|
|
|
GE_IF_BOOL_EXEC(!op_desc->SetExtAttr(kRemainNode, true), GELOGE(INTERNAL_ERROR, "set ext attr failed"); return); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
@@ -594,7 +594,7 @@ void TransOpWithoutReshapeFusionPass::GetBeginOutDescAndEndInDesc(const int inde |
|
|
|
auto out_owner_node = out_peer_anchor->GetOwnerNode(); |
|
|
|
GE_CHECK_NOTNULL_JUST_RETURN(out_owner_node); |
|
|
|
auto out_peer_op_desc = out_owner_node->GetOpDesc(); |
|
|
|
GE_IF_BOOL_EXEC(out_peer_op_desc == nullptr, GELOGE(INTERNAL_ERROR, "out_peer_op_desc is nullptr"); return ); |
|
|
|
GE_IF_BOOL_EXEC(out_peer_op_desc == nullptr, GELOGE(INTERNAL_ERROR, "out_peer_op_desc is nullptr"); return); |
|
|
|
out_desc = out_peer_op_desc->GetInputDesc(out_peer_anchor->GetIdx()); |
|
|
|
|
|
|
|
auto in_peer_anchor = nodes_anchor.back().first; |
|
|
@@ -602,7 +602,7 @@ void TransOpWithoutReshapeFusionPass::GetBeginOutDescAndEndInDesc(const int inde |
|
|
|
auto in_owner_node = in_peer_anchor->GetOwnerNode(); |
|
|
|
GE_CHECK_NOTNULL_JUST_RETURN(in_owner_node); |
|
|
|
auto in_peer_op_desc = in_owner_node->GetOpDesc(); |
|
|
|
GE_IF_BOOL_EXEC(in_peer_op_desc == nullptr, GELOGE(INTERNAL_ERROR, "in_peer_op_desc is nullptr"); return ); |
|
|
|
GE_IF_BOOL_EXEC(in_peer_op_desc == nullptr, GELOGE(INTERNAL_ERROR, "in_peer_op_desc is nullptr"); return); |
|
|
|
in_desc = in_peer_op_desc->GetOutputDesc(in_peer_anchor->GetIdx()); |
|
|
|
} |
|
|
|
|
|
|
@@ -734,7 +734,7 @@ void TransOpWithoutReshapeFusionPass::RemoveNousedNodes(const ComputeGraphPtr &g |
|
|
|
continue; |
|
|
|
} |
|
|
|
|
|
|
|
GE_IF_BOOL_EXEC(!op_desc->SetExtAttr(kRemainNode, true), GELOGE(INTERNAL_ERROR, "set ext attr failed"); return ); |
|
|
|
GE_IF_BOOL_EXEC(!op_desc->SetExtAttr(kRemainNode, true), GELOGE(INTERNAL_ERROR, "set ext attr failed"); return); |
|
|
|
GELOGI("remove node:%s", node->GetName().c_str()); |
|
|
|
if (graph->RemoveNode(node) != GRAPH_SUCCESS) { |
|
|
|
GELOGW("remove node failed!node:%s", node->GetName().c_str()); |
|
|
|