diff --git a/ge/graph/passes/transpose_transdata_pass.cc b/ge/graph/passes/transpose_transdata_pass.cc index 0f3e7e70..810f5639 100644 --- a/ge/graph/passes/transpose_transdata_pass.cc +++ b/ge/graph/passes/transpose_transdata_pass.cc @@ -86,7 +86,7 @@ Status TransposeTransDataPass::Run(NodePtr &node) { if (CheckOneInAndOneOutDataAnchor(out_node)) { return FAILED; } - if (!FusionIfNeed(node, out_op_desc)) { + if (!FusionIfNeed(op_desc, out_node)) { continue; } CopyInputEdges(node, out_node); @@ -152,8 +152,8 @@ Status TransposeTransDataPass::RemoveTranspose(NodePtr &node) { return SUCCESS; } -bool TransposeTransDataPass::FusionIfNeed(NodePtr &node, OpDescPtr &transdata_op_desc) { - auto op_desc = node->GetOpDesc(); +bool TransposeTransDataPass::FusionIfNeed(OpDescPtr &op_desc, NodePtr &node) { + auto transdata_op_desc = node->GetOpDesc(); GE_CHECK_NOTNULL(op_desc); GE_CHECK_NOTNULL(transdata_op_desc); auto out_input_desc = transdata_op_desc->MutableInputDesc(0); diff --git a/ge/graph/passes/transpose_transdata_pass.h b/ge/graph/passes/transpose_transdata_pass.h index ce42ba39..c6ef0b36 100644 --- a/ge/graph/passes/transpose_transdata_pass.h +++ b/ge/graph/passes/transpose_transdata_pass.h @@ -26,7 +26,7 @@ class TransposeTransDataPass : public BaseNodePass { private: Status CheckOneInAndOneOutDataAnchor(NodePtr &node) const; Status RemoveTranspose(NodePtr &node); - bool FusionIfNeed(NodePtr &node, OpDescPtr &transdata_op_desc); + bool FusionIfNeed(OpDescPtr &op_desc, NodePtr &node); void CopyInputEdges(NodePtr &origin_node, NodePtr &new_node); bool TransDataCheckAccuracySupported(NodePtr &node); };