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