From 1690fa5c70d17e404e2571bb62fa4a715fc92dba Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Mon, 18 Jan 2021 17:29:31 +0800 Subject: [PATCH] modified: tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc --- tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc b/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc index 068ef5f4..d5229b2c 100644 --- a/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc +++ b/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc @@ -22,6 +22,7 @@ #include "common/util.h" #include "graph/passes/graph_builder_utils.h" #include "graph/utils/attr_utils.h" +#include "graph/debug/ge_attr_define.h" #define private public #define protected public @@ -65,7 +66,6 @@ TEST_F(UtestGraphPreproces, test_dynamic_input_shape_parse) { EXPECT_EQ(ret, ge::SUCCESS); // check data node output shape_range and shape auto data_node = graph_prepare.compute_graph_->FindNode("data1"); -) auto data_output_desc = data_node->GetOpDesc()->GetOutputDescPtr(0); vector expect_shape = {3,-1,-1,5}; auto result_shape = data_output_desc->GetShape();