diff --git a/ge/generator/ge_generator.cc b/ge/generator/ge_generator.cc
index 7a489bfd..2baa2c5b 100644
--- a/ge/generator/ge_generator.cc
+++ b/ge/generator/ge_generator.cc
@@ -269,14 +269,12 @@ static void ResetTensorVecShape(const vector<GeTensor> &inputs, vector<GeTensor>
   for (auto input : inputs) {
     auto input_desc = input.GetTensorDesc();
     GeShape shape_ori = input_desc.GetShape();
-    Format format_ori = input_desc.GetFormat();
-    DataType type_ori = input_desc.GetDataType();
 
     std::vector<int64_t> dynamic_shape_dims = {kDynamicDimValue};
     GeShape dynamic_shape(dynamic_shape_dims);
 
     ge::GeTensor inputTensor;
-    ge::GeTensorDesc desc(shape_ori, format_ori, type_ori);
+    ge::GeTensorDesc desc(input_desc);
     if (shape_ori.GetDims().size() > 0) {
       desc.SetShape(dynamic_shape);
     }