Browse Source

Feature: Get default from ge ir graph while no user input shape

tags/v1.2.0
l00444296 4 years ago
parent
commit
30dbe7d97f
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      ge/ir_build/ge_ir_build.cc

+ 1
- 0
ge/ir_build/ge_ir_build.cc View File

@@ -310,6 +310,7 @@ graphStatus Impl::GetDefaultInputShape(const Graph &graph, string &default_shape
GELOGD("Data op name: %s, data shape: %s", data_op_name.c_str(), tmp_shape_str.c_str());
}
}
default_shape = (default_shape.empty() ? default_shape : default_shape.substr(0, default_shape.size() - 1));
GELOGI("Get default data op shape from ge ir graph: %s", default_shape.c_str());
return GRAPH_SUCCESS;
}


Loading…
Cancel
Save