diff --git a/ge/graph/passes/infershape_pass.cc b/ge/graph/passes/infershape_pass.cc index 0555929d..bab07403 100755 --- a/ge/graph/passes/infershape_pass.cc +++ b/ge/graph/passes/infershape_pass.cc @@ -56,10 +56,9 @@ void UpdateShapeAndDType(const GeTensorDescPtr &src, GeTensorDescPtr &dst) { dst->SetShape(src->GetShape()); dst->SetDataType(src->GetDataType()); dst->SetOriginDataType(src->GetOriginDataType()); - vector> src_shape_range; - src->GetShapeRange(src_shape_range); - dst->SetShapeRange(src_shape_range); - dst->SetOriginShapeRange(src_shape_range); + vector> dst_shape_range; + dst->GetShapeRange(dst_shape_range); + dst->SetOriginShapeRange(dst_shape_range); ge::TensorUtils::SetRealDimCnt(*dst, static_cast(src->GetShape().GetDims().size())); } } // namespace