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

[LPT] Fix out of bounds access in reshape (#10791)

Browse Source
This commit is contained in:
Vladimir Zinoviev 2022-03-11 18:04:14 +03:00 committed by GitHub
parent 86322c916b
commit eebe8c70f9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/common/low_precision_transformations/src/reshape.cpp
View File

@ -238,18 +238,20 @@ bool ReshapeTransformation::canBeTransformed(const TransformationContext& contex
multiplyShapeWithBatch.insert(multiplyShapeWithBatch.begin(), 1ul);
}
const size_t outputChannel = static_cast<size_t>(outputPShape[1].get_length());
if (!subtractShapeWithBatch.empty() && (outputChannel < subtractShapeWithBatch[1])) {
return false;
}
if (!multiplyShapeWithBatch.empty() && (outputChannel < multiplyShapeWithBatch[1])) {
return false;
}
if (subtractShapeWithBatch.size() > 1 && multiplyShapeWithBatch.size() > 1) {
const size_t outputChannel = static_cast<size_t>(outputPShape[1].get_length());
if (!subtractShapeWithBatch.empty() && (outputChannel < subtractShapeWithBatch[1])) {
return false;
}
if (!multiplyShapeWithBatch.empty() && (outputChannel < multiplyShapeWithBatch[1])) {
return false;
}
if (outputPShape.is_static() &&
((!subtractShapeWithBatch.empty() && ((outputChannel % subtractShapeWithBatch[1]) != 0)) ||
(!multiplyShapeWithBatch.empty() && (outputChannel % multiplyShapeWithBatch[1] != 0)))) {
return false;
if (outputPShape.is_static() &&
((!subtractShapeWithBatch.empty() && ((outputChannel % subtractShapeWithBatch[1]) != 0)) ||
(!multiplyShapeWithBatch.empty() && (outputChannel % multiplyShapeWithBatch[1] != 0)))) {
return false;
}
}
return canBeTransformed(subtractShapeWithBatch, multiplyShapeWithBatch, inputPShape, outputPShape);