m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modified: tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc

Browse Source
pull/1157/head
zhaoxinxin 5 years ago
parent ce359a3c32
commit 554d08f277
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
tests/ut/ge/graph/passes/replace_with_empty_const_pass_unittest.cc
Unescape View File

@ -75,7 +75,7 @@ TEST_F(UtestReplaceWithEmptyConstPass, replace_whith_empty_const_success) {
EXPECT_EQ(graph->GetDirectNodesSize(),5); EXPECT_EQ(graph->GetDirectNodesSize(),5);
auto cast1 = graph->FindNode("cast1"); auto cast1 = graph->FindNode("cast1");
ret = replace_with_empty_const_pass.Run(cast1) ret = replace_with_empty_const_pass.Run(cast1);
EXPECT_EQ(cast1->GetOutAllNodes().size(),0); EXPECT_EQ(cast1->GetOutAllNodes().size(),0);
auto conv2d = graph->FindNode("conv2d"); auto conv2d = graph->FindNode("conv2d");
EXPECT_EQ(conv2d->GetInDataNodes().at(0)->GetType(),"Const"); EXPECT_EQ(conv2d->GetInDataNodes().at(0)->GetType(),"Const");

Write Preview
Loading…
Cancel
Save