|
|
|
@ -269,6 +269,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
|
|
|
|
|
NodePtr sub_sub_graph_while_body_const_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_const_op_desc);*/
|
|
|
|
|
OpDescPtr sub_sub_graph_while_body_data_op_desc = CreateOpDesc("body_data", DATA);
|
|
|
|
|
NodePtr sub_sub_graph_while_body_data_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_data_op_desc);
|
|
|
|
|
sub_sub_graph2->SetGraphUnknownFlag(true);
|
|
|
|
|
/*OpDescPtr sub_sub_graph_while_body_add_op_desc = CreateOpDesc("body_add", ADD);
|
|
|
|
|
NodePtr sub_sub_graph_while_body_add_node = sub_sub_graph2->AddNode(sub_sub_graph_while_body_add_node);
|
|
|
|
|
sub_sub_graph_while_body_add_node->AddLinkFrom(sub_sub_graph_while_body_data_node);
|
|
|
|
@ -277,7 +278,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
|
|
|
|
|
ComputeGraphPtr sub_graph = std::make_shared<ComputeGraph>("sub_graph");
|
|
|
|
|
OpDescPtr sub_graph_while_op_desc = CreateOpDesc("while", WHILE);
|
|
|
|
|
NodePtr sub_graph_while_node = sub_graph->AddNode(sub_graph_while_op_desc);
|
|
|
|
|
sub_graph->SetGraphUnknownFlag(true);
|
|
|
|
|
sub_graph->SetGraphUnknownFlag(false);
|
|
|
|
|
sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_cond");
|
|
|
|
|
sub_graph_while_node->GetOpDesc()->AddSubgraphName("while_body");
|
|
|
|
|
sub_graph_while_node->GetOpDesc()->SetSubgraphInstanceName(0, "while_cond");
|
|
|
|
@ -289,6 +290,7 @@ TEST_F(UtestGeHybrid, unfold_subgraphs_success) {
|
|
|
|
|
partitioned_call_node->GetOpDesc()->AddSubgraphName("sub_graph");
|
|
|
|
|
partitioned_call_node->GetOpDesc()->SetSubgraphInstanceName(0, "sub_graph");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
root_graph->AddSubGraph(sub_sub_graph1);
|
|
|
|
|
root_graph->AddSubGraph(sub_sub_graph2);
|
|
|
|
|
sub_sub_graph1->SetParentGraph(root_graph);
|
|
|
|
|