| @@ -259,10 +259,9 @@ static void ResetInputShape(const vector<GeTensor> &inputs, vector<GeTensor> &in | |||||
| GeShape dynamic_shape(dynamic_shape_dims); | GeShape dynamic_shape(dynamic_shape_dims); | ||||
| ge::GeTensor inputTensor; | 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); | 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, | const string &model_file_name, OpEngineType engine_type, ModelBufferData &model_buff, | ||||
| bool is_offline) { | bool is_offline) { | ||||
| if (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) { | if (CheckForSingleOp(op_desc, inputs, outputs) != SUCCESS) { | ||||
| GELOGE(PARAM_INVALID, "input param is invalid when build single op!"); | GELOGE(PARAM_INVALID, "input param is invalid when build single op!"); | ||||