modified: ge/graph/passes/transop_without_reshape_fusion_pass.cc

pull/167/head
zhaoxinxin 4 years ago
parent 61d3f45e9a
commit 9bde140ac0

@ -131,7 +131,7 @@ graphStatus TransOpWithoutReshapeFusionPass::GetSubGraphNodesInfo() {
sub_graph_has_reshape_node[i] = true;
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 output_format = in_node->GetOpDesc()->GetOutputDescPtr(0)->GetFormat();
if (input_format == output_format) {

Loading…
Cancel
Save