Browse Source

!506 us/st fix

Merge pull request !506 from 徐睿/ge_dev
pull/505/head
i-robot Gitee 3 years ago
parent
commit
1fc4182275
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      tests/st/testcase/test_tensorflow_parser.cc
  2. +1
    -1
      tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc

+ 1
- 1
tests/st/testcase/test_tensorflow_parser.cc View File

@@ -1124,7 +1124,7 @@ TEST_F(STestTensorflowParser, tensorflow_parserfrommemory_failed)
ret = ge::aclgrphParseTensorFlow(modelFile.c_str(), parser_params, graph);
ge::ComputeGraphPtr compute_graph = ge::GraphUtils::GetComputeGraph(graph);
ret = modelParser.ParseFromMemory(data, size, compute_graph);
EXPECT_EQ(ret, INTERNAL_ERROR);
EXPECT_NE(ret, SUCCESS);
}

TEST_F(STestTensorflowParser, modelparser_parsefrommemory_success)


+ 1
- 1
tests/ut/parser/testcase/tensorflow_parser_testcase/tensorflow_parser_unittest.cc View File

@@ -1288,7 +1288,7 @@ TEST_F(UtestTensorflowParser, tensorflow_parserfrommemory_failed)
ret = ge::aclgrphParseTensorFlow(modelFile.c_str(), parser_params, graph);
ge::ComputeGraphPtr compute_graph = ge::GraphUtils::GetComputeGraph(graph);
ret = modelParser.ParseFromMemory(data, size, compute_graph);
EXPECT_EQ(ret, INTERNAL_ERROR);
EXPECT_NE(ret, SUCCESS);
}

TEST_F(UtestTensorflowParser, modelparser_parsefrommemory_success)


Loading…
Cancel
Save