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

[LPT] Turn back checks in reshape transformation when subtract is absent (#10939)

Browse Source
This commit is contained in:
Vladimir Zinoviev 2022-03-15 11:34:12 +03:00 committed by GitHub
parent ef5ad90dd7
commit 4a2d0f39dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 13 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

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