|
|
|
@@ -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<int64_t> expect_shape = {3,-1,-1,5}; |
|
|
|
auto result_shape = data_output_desc->GetShape(); |
|
|
|
|