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