[CPU] add check on equality of values IL and IH ranges (#2190)

This commit is contained in:
Maxim Andronov 2020-09-14 15:28:37 +03:00 committed by GitHub
parent 706d2fe7c8
commit 8cf84f58e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -160,6 +160,13 @@ void MKLDNNQuantizeNode::init() {
break;
}
}
for (ptrdiff_t i = 0; i < std::max(inputLowAxisSize, inputHighAxisSize); i++) {
if (inputLowData[isInputLowBroadcasted ? 0 : i] != inputHighData[isInputHighBroadcasted ? 0 : i]) {
binarization = false;
break;
}
}
}
if (binarization) {