Browse Source

Fix bug.

pull/1190/head
unknown 4 years ago
parent
commit
9bf5113af6
2 changed files with 4 additions and 4 deletions
  1. +3
    -3
      ge/graph/passes/transpose_transdata_pass.cc
  2. +1
    -1
      ge/graph/passes/transpose_transdata_pass.h

+ 3
- 3
ge/graph/passes/transpose_transdata_pass.cc View File

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


+ 1
- 1
ge/graph/passes/transpose_transdata_pass.h View File

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


Loading…
Cancel
Save