From 157b8c4a0a96f8ce691cc180df36dd0799cfa132 Mon Sep 17 00:00:00 2001 From: "River.Li" Date: Sat, 17 Jun 2023 10:39:18 +0800 Subject: [PATCH] Fix minor issue --- src/inference/src/blob_transform.cpp | 2 +- src/inference/src/dev/make_tensor.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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;