From 74acea8cd13ef1f7dc309f31d1a3fcbb2be3dc6e Mon Sep 17 00:00:00 2001 From: zhaoxinxin Date: Mon, 19 Jul 2021 20:19:24 +0800 Subject: [PATCH] fix duzzy compile set input origin shape range --- ge/graph/passes/infershape_pass.cc | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) 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