From 56649dcc8136f942330d3ae4214ed82339e1054b Mon Sep 17 00:00:00 2001
From: l00444296 <lixiwen1@huawei.com>
Date: Thu, 10 Dec 2020 14:48:00 +0800
Subject: [PATCH] Feature: reset shape of dynamic single op

---
 ge/generator/ge_generator.cc | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index 614b69ef..7140170e 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -259,10 +259,9 @@ static void ResetInputShape(const vector<GeTensor> &inputs, vector<GeTensor> &in
     GeShape dynamic_shape(dynamic_shape_dims);
 
     ge::GeTensor inputTensor;
-    if (shape_ori.GetDims().size() == 0) {
-      ge::GeTensorDesc desc(shape_ori, format_ori, type_ori);
-    } else {
-      ge::GeTensorDesc desc(dynamic_shape, format_ori, type_ori);
+    ge::GeTensorDesc desc(shape_ori, format_ori, type_ori);
+    if (shape_ori.GetDims().size() > 0) {
+      desc.SetShape(dynamic_shape);
     }
 
     inputTensor.SetTensorDesc(desc);
@@ -597,7 +596,7 @@ Status GeGenerator::BuildSingleOp(OpDescPtr &op_desc, const vector<GeTensor> &in
                                   const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff,
                                   bool is_offline) {
   if (is_offline) {
-    (void)AttrUtils::SetBool(data_op, ATTR_DYNAMIC_SHAPE_SINGLE_AICPU, true);
+    (void)AttrUtils::SetBool(op_desc, ATTR_DYNAMIC_SHAPE_SINGLE_AICPU, true);
   }
   if (CheckForSingleOp(op_desc, inputs, outputs) != SUCCESS) {
     GELOGE(PARAM_INVALID, "input param is invalid when build single op!");