[CPU] add check on equality of values IL and IH ranges (#2190)
This commit is contained in:
parent
706d2fe7c8
commit
8cf84f58e7
@ -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) {
|
||||
|
Loading…
Reference in New Issue
Block a user