Browse Source

dynamic shape inference support

pull/1240/head
lichun 4 years ago
parent
commit
645f2abfee
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      tests/ut/ge/generator/ge_generator_unittest.cc

+ 1
- 1
tests/ut/ge/generator/ge_generator_unittest.cc View File

@@ -76,7 +76,7 @@ TEST_F(UtestGeGenerator, test_build_single_op_online) {
TEST_F(UtestGeGenerator, test_set_model_name) {
GeGenerator generator;
generator.Initialize({});
GeRootModelPtr ge_root_model = make_shared<GeRootModelPtr>(GeRootModel());
GeRootModelPtr ge_root_model = make_shared<GeRootModel>(GeRootModel());
ComputeGraphPtr graph = make_shared<ComputeGraph>(ComputeGraph("graph"));
(void)AttrUtils::SetBool(graph, "_dynamic_shape_partitioned", true);
ge_root_model->root_graph_ = std::move(graph);


Loading…
Cancel
Save