Browse Source

fix duzzy compile set input origin shape range

pull/2047/head
zhaoxinxin 3 years ago
parent
commit
74acea8cd1
1 changed files with 3 additions and 4 deletions
  1. +3
    -4
      ge/graph/passes/infershape_pass.cc

+ 3
- 4
ge/graph/passes/infershape_pass.cc View File

@@ -56,10 +56,9 @@ void UpdateShapeAndDType(const GeTensorDescPtr &src, GeTensorDescPtr &dst) {
dst->SetShape(src->GetShape());
dst->SetDataType(src->GetDataType());
dst->SetOriginDataType(src->GetOriginDataType());
vector<pair<int64_t, int64_t>> src_shape_range;
src->GetShapeRange(src_shape_range);
dst->SetShapeRange(src_shape_range);
dst->SetOriginShapeRange(src_shape_range);
vector<pair<int64_t, int64_t>> dst_shape_range;
dst->GetShapeRange(dst_shape_range);
dst->SetOriginShapeRange(dst_shape_range);
ge::TensorUtils::SetRealDimCnt(*dst, static_cast<uint32_t>(src->GetShape().GetDims().size()));
}
} // namespace


Loading…
Cancel
Save