From 30dbe7d97fc05586133f29a2e1229579bd551327 Mon Sep 17 00:00:00 2001 From: l00444296 Date: Tue, 8 Dec 2020 14:56:26 +0800 Subject: [PATCH] Feature: Get default from ge ir graph while no user input shape --- ge/ir_build/ge_ir_build.cc | 1 + 1 file changed, 1 insertion(+) 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; }