IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix heap-buffer-overflow issue. (#14648)

Browse Source
This commit is contained in:
cecilia peng 2022-12-21 17:12:08 +08:00 committed by GitHub
parent 3cf64c329e
commit 2d5fa2d164
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/frontends/paddle/src/input_model.cpp
View File

@ -364,7 +364,7 @@ void InputModel::InputModelImpl::createTempConsts() {
var_place->set_element_type(type); var_place->set_element_type(type);
var_place->set_partial_shape(tensor_ps); var_place->set_partial_shape(tensor_ps);
Shape shape(tensor_ps.size()); Shape shape(tensor_ps.size(), 0);
for (auto i = 0; i < tensor_ps.size(); i++) { for (auto i = 0; i < tensor_ps.size(); i++) {
const auto& dim = tensor_ps[i]; const auto& dim = tensor_ps[i];
if (dim.is_static()) { if (dim.is_static()) {
@ -373,7 +373,10 @@ void InputModel::InputModelImpl::createTempConsts() {
} }
if (tensor_ps.is_static()) { 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}); auto node = opset7::Constant::create(type, shape, {0});