|
|
|
@@ -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); |
|
|
|
|