|
@@ -17,6 +17,10 @@ |
|
|
#include <gtest/gtest.h> |
|
|
#include <gtest/gtest.h> |
|
|
#include "graph/partition/dynamic_shape_partition.h" |
|
|
#include "graph/partition/dynamic_shape_partition.h" |
|
|
#include "compute_graph.h" |
|
|
#include "compute_graph.h" |
|
|
|
|
|
#include "inc/framework/common/types.h" |
|
|
|
|
|
#include "utils/graph_utils.h" |
|
|
|
|
|
#include "graph/debug/ge_attr_define.h" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#define private public |
|
|
#define private public |
|
|
#define protected public |
|
|
#define protected public |
|
@@ -75,7 +79,7 @@ class UtestDynamicShapePartition : public testing::Test { |
|
|
|
|
|
|
|
|
// test Init_EndGraphTaskInfo_failed |
|
|
// test Init_EndGraphTaskInfo_failed |
|
|
TEST_F(UtestDynamicShapePartition, single_op_scene_success) { |
|
|
TEST_F(UtestDynamicShapePartition, single_op_scene_success) { |
|
|
ComputeGraphPtr computeGraph("default"); |
|
|
|
|
|
|
|
|
ComputeGraphPtr graph = shared_ptr<ComputeGraph>("default"); |
|
|
|
|
|
|
|
|
NodePtr node1 = |
|
|
NodePtr node1 = |
|
|
NodeBuilder("node1", CONSTANTOP).AddInputDesc({1, 1, 224, 224}).AddOutputDesc({1, 1, 224, 224}).Build(graph); |
|
|
NodeBuilder("node1", CONSTANTOP).AddInputDesc({1, 1, 224, 224}).AddOutputDesc({1, 1, 224, 224}).Build(graph); |
|
|