Browse Source

use true types istead of GeAttrValue::

pull/372/head
CLAY-panjw 4 years ago
parent
commit
d57c76994d
1 changed files with 6 additions and 6 deletions
  1. +6
    -6
      tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc

+ 6
- 6
tests/ut/parser/testcase/graph_optimizer_testcase/graph_optimizer_unittest.cc View File

@@ -52,7 +52,7 @@ namespace {
}
}

TEST_F(UtestGraphOptimizer, graph_optimizer) {
TEST_F(UtestGraphOptimizer, graph_optimizer) {
ge::ComputeGraphPtr graph = MakeGraph();
ge::IteratorFusionPass iteratorFusionPass(domi::TENSORFLOW);
EXPECT_NE(iteratorFusionPass.Run(graph),ge::SUCCESS);
@@ -61,11 +61,11 @@ TEST_F(UtestGraphOptimizer, graph_optimizer) {
TEST_F(UtestGraphOptimizer, graph_optimizer_output) {
ge::ComputeGraphPtr graph = MakeGraph();
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);
}
}

Loading…
Cancel
Save