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

[CPU] add const and precision check quantize ranges (#2074)

Browse Source
This commit is contained in:
Maxim Andronov 2020-09-09 18:57:58 +03:00 committed by GitHub
parent 3af3e459c3
commit b3829c58a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
inference-engine/src/mkldnn_plugin/nodes/mkldnn_quantize_node.cpp
View File

@ -124,6 +124,14 @@ void MKLDNNQuantizeNode::init() {
THROW_IE_EXCEPTION << "Unsupported input sizes for Quantize layer with name " << getName();
}
for (size_t i = 1; i < getParentEdges().size(); i++) {
if (!getParentEdgesAtPort(i)[0]->getParent()->isConstant())
THROW_IE_EXCEPTION << "Quantize layer with name " << getName() << " has non const input on " << i << " port";
auto prec = getCnnLayer()->insData[i].lock()->getPrecision();
if (prec != Precision::FP32)
THROW_IE_EXCEPTION << "Quantize layer with name " << getName() << " has unsupported precision " << prec << " on " << i << " port";
}
auto inputLowBlob = dynamic_cast<TBlob<float>*>(getParentEdgesAtPort(1)[0]->getParent()->getCnnLayer()->blobs["custom"].get());
auto inputLowData = inputLowBlob->buffer().as<float*>();