From e274ac6a1c96156b87aa715843c2ac2d8a07be5d Mon Sep 17 00:00:00 2001 From: "mei, yang" Date: Sun, 24 Mar 2024 23:17:32 +0800 Subject: [PATCH] fix heap-buffer-overflow in paddle frontend (#23599) ### Details: - *fix heap-buffer-overflow in paddle frontend* ### Tickets: - *CVS-135869* --- src/frontends/paddle/src/input_model.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/frontends/paddle/src/input_model.cpp b/src/frontends/paddle/src/input_model.cpp index 372ad63f434..9b2cc6bfdc6 100644 --- a/src/frontends/paddle/src/input_model.cpp +++ b/src/frontends/paddle/src/input_model.cpp @@ -451,7 +451,10 @@ void InputModel::InputModelImpl::createTempConsts() { } if (tensor_ps.is_static()) { - shape[1] = 0; + // this tensorarray tensor originally could be scalar, then + // tensor_ps size would be 1 after unsqueeze. + auto idx = tensor_ps.size() > 1 ? 1 : 0; + shape[idx] = 0; } auto node = opset7::Constant::create(type, shape, {0});