| @@ -52,7 +52,7 @@ namespace { | |||||
| } | } | ||||
| } | } | ||||
| TEST_F(UtestGraphOptimizer, graph_optimizer) { | |||||
| TEST_F(UtestGraphOptimizer, graph_optimizer) { | |||||
| ge::ComputeGraphPtr graph = MakeGraph(); | ge::ComputeGraphPtr graph = MakeGraph(); | ||||
| ge::IteratorFusionPass iteratorFusionPass(domi::TENSORFLOW); | ge::IteratorFusionPass iteratorFusionPass(domi::TENSORFLOW); | ||||
| EXPECT_NE(iteratorFusionPass.Run(graph),ge::SUCCESS); | EXPECT_NE(iteratorFusionPass.Run(graph),ge::SUCCESS); | ||||
| @@ -61,11 +61,11 @@ TEST_F(UtestGraphOptimizer, graph_optimizer) { | |||||
| TEST_F(UtestGraphOptimizer, graph_optimizer_output) { | TEST_F(UtestGraphOptimizer, graph_optimizer_output) { | ||||
| ge::ComputeGraphPtr graph = MakeGraph(); | ge::ComputeGraphPtr graph = MakeGraph(); | ||||
| domi::FrameworkType type = domi::TENSORFLOW; | domi::FrameworkType type = domi::TENSORFLOW; | ||||
| ge::ParserGraphOptimizer parserGraphOptimizer(graph,type); | |||||
| vector<ge::InDataAnchorPtr> input_anchors; | |||||
| vector<ge::OutDataAnchorPtr> output_anchors; | |||||
| ge::OpDescPtr fusion_op_desc; | |||||
| EXPECT_NE(parserGraphOptimizer.RebuildInputAnchors(input_anchors,fusion_op_desc),ge::SUCCESS); | |||||
| ge::ParserGraphOptimizer parserGraphOptimizer(graph,type); | |||||
| vector<ge::InDataAnchorPtr> input_anchors; | |||||
| vector<ge::OutDataAnchorPtr> output_anchors; | |||||
| ge::OpDescPtr fusion_op_desc; | |||||
| EXPECT_NE(parserGraphOptimizer.RebuildInputAnchors(input_anchors,fusion_op_desc),ge::SUCCESS); | |||||
| EXPECT_NE(parserGraphOptimizer.RebuildOutputAnchors(output_anchors,fusion_op_desc),ge::SUCCESS); | EXPECT_NE(parserGraphOptimizer.RebuildOutputAnchors(output_anchors,fusion_op_desc),ge::SUCCESS); | ||||
| } | } | ||||
| } | } | ||||