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

[LPT] Fixed quantizeBlob routine for 3D case

Browse Source
This commit is contained in:
dmitrygo 2020-06-09 09:37:03 +03:00 committed by Alexander Peskov
parent 8eb88d51f2
commit 051a429c31
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/low_precision_transformations/src/network_helper.cpp
View File

@ -1734,7 +1734,7 @@ void CNNNetworkHelper::quantizeBlob(const CNNLayer& quantize, Blob::Ptr& targetB
const size_t OC = outDims[0];
const size_t IC = outDims.size() > 1lu ? outDims[1] : 1;
const size_t D = outDims.size() > 4lu ? outDims[outDims.size() - 3] : 1;
const size_t H = outDims.size() > 2lu ? outDims[outDims.size() - 2] : 1;
const size_t H = outDims.size() > 2lu ? outDims.size() == 3lu ? outDims[2] : outDims[outDims.size() - 2] : 1;
const size_t W = outDims.size() > 3lu ? outDims[outDims.size() - 1] : 1;
// Const layer blob shape (sourceBlob->getTensorDesc().getDims()) can be different from output port shape