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 (#10940)

Browse Source
This commit is contained in:
Vladimir Zinoviev
2022-03-15 11:34:05 +03:00
committed by GitHub
parent a423a2b802
commit b4b5f3333e
1 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -238,20 +238,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);