From 051a429c311b10ca8ac34e996fadaada7292ed7a Mon Sep 17 00:00:00 2001 From: dmitrygo Date: Tue, 9 Jun 2020 09:37:03 +0300 Subject: [PATCH] [LPT] Fixed quantizeBlob routine for 3D case --- .../src/low_precision_transformations/src/network_helper.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/src/low_precision_transformations/src/network_helper.cpp b/inference-engine/src/low_precision_transformations/src/network_helper.cpp index 5318ebaf4d8..922f7a630ad 100644 --- a/inference-engine/src/low_precision_transformations/src/network_helper.cpp +++ b/inference-engine/src/low_precision_transformations/src/network_helper.cpp @@ -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