Browse Source

Eliminate variable_op_list_

tags/v1.2.0
zhangxiaokun 4 years ago
parent
commit
501111b020
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      tests/ut/ge/graph/load/davinci_model_unittest.cc

+ 4
- 1
tests/ut/ge/graph/load/davinci_model_unittest.cc View File

@@ -308,6 +308,9 @@ TEST_F(UtestDavinciModel, init_unknown) {

TEST_F(UtestDavinciModel, Init_variable_op) {
DavinciModel model(0, nullptr);
model.ge_model_ = make_shared<GeModel>();
model.runtime_param_.mem_base = (uint8_t *)0x08000000;
model.runtime_param_.mem_size = 5120000;
ComputeGraphPtr graph = make_shared<ComputeGraph>("default");

GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
@@ -371,7 +374,7 @@ TEST_F(UtestDavinciModel, InitRealSizeAndShapeInfo_succ2) {

GraphUtils::AddEdge(data1_node->GetOutDataAnchor(0), case1_node->GetInDataAnchor(0));
GraphUtils::AddEdge(case1_node->GetOutDataAnchor(0), output_node->GetInDataAnchor(0));
(void)AttrUtils::SetStr(output_node->GetOpDesc(), ATTR_ALL_GEARS_INFO, "1;2;4;8");
(void)AttrUtils::SetBool(case_node, ATTR_INSERT_BY_MBATCH, true);



Loading…
Cancel
Save