m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modified: ge/graph/passes/transop_without_reshape_fusion_pass.cc

Browse Source
pull/167/head
zhaoxinxin 4 years ago
parent 61d3f45e9a
commit 9bde140ac0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
ge/graph/passes/transop_without_reshape_fusion_pass.cc
Unescape View File

@ -131,7 +131,7 @@ graphStatus TransOpWithoutReshapeFusionPass::GetSubGraphNodesInfo() {
sub_graph_has_reshape_node[i] = true; sub_graph_has_reshape_node[i] = true;
break; break;
} }
if (in_node->GetType() == TRANSOPSE || in_node->GetType() == TRANSOPSED) { if (in_node->GetType() == TRANSPOSE || in_node->GetType() == TRANSPOSED) {
auto input_format = in_node->GetOpDesc()->GetInputDescPtr(0)->GetFormat(); auto input_format = in_node->GetOpDesc()->GetInputDescPtr(0)->GetFormat();
auto output_format = in_node->GetOpDesc()->GetOutputDescPtr(0)->GetFormat(); auto output_format = in_node->GetOpDesc()->GetOutputDescPtr(0)->GetFormat();
if (input_format == output_format) { if (input_format == output_format) {

Write Preview
Loading…
Cancel
Save