diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc index fbacf527..2035dbe0 100644 --- a/ge/ir_build/ge_ir_build.cc +++ b/ge/ir_build/ge_ir_build.cc @@ -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; }