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