diff --git a/src/inference/src/blob_transform.cpp b/src/inference/src/blob_transform.cpp index 75fafaf077f..0ccbe03a388 100644 --- a/src/inference/src/blob_transform.cpp +++ b/src/inference/src/blob_transform.cpp @@ -363,7 +363,7 @@ void blob_copy(Blob::Ptr src, Blob::Ptr dst) { if (src->getTensorDesc().getPrecision() != dst->getTensorDesc().getPrecision()) IE_THROW() << "Unimplemented blob transformation from precision " << src->getTensorDesc().getPrecision() - << " to " << src->getTensorDesc().getPrecision(); + << " to " << dst->getTensorDesc().getPrecision(); if (src->getTensorDesc().getDims() != dst->getTensorDesc().getDims()) IE_THROW() << "Unimplemented blob transformation from different shapes "; diff --git a/src/inference/src/dev/make_tensor.cpp b/src/inference/src/dev/make_tensor.cpp index ff8b3b507e6..77e3ea610fb 100644 --- a/src/inference/src/dev/make_tensor.cpp +++ b/src/inference/src/dev/make_tensor.cpp @@ -512,12 +512,12 @@ public: auto dims = shape; if (shape.size() == 4) { // NHWC - if (shape[1] == shape[2]) { + if ((shape[1] == shape[2]) && (shape[1] != shape[3])) { blk_order = {0, 2, 3, 1}; dims[1] = shape[3]; dims[2] = shape[1]; dims[3] = shape[2]; - } + } } ie::SizeVector dim_offset(shape.size(), 0); ie::SizeVector blk_strides;