Eliminate variable_op_list_

pull/880/head
zhangxiaokun 4 years ago
parent b0cbef78fc
commit 501111b020

@ -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