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

Fix minor issue

Browse Source
This commit is contained in:
River.Li 2023-06-17 10:39:18 +08:00
parent 48fd6296b5
commit 157b8c4a0a
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/inference/src/blob_transform.cpp
View File

@ -363,7 +363,7 @@ void blob_copy(Blob::Ptr src, Blob::Ptr dst) {
if (src->getTensorDesc().getPrecision() != dst->getTensorDesc().getPrecision()) if (src->getTensorDesc().getPrecision() != dst->getTensorDesc().getPrecision())
IE_THROW() << "Unimplemented blob transformation from precision " << src->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()) if (src->getTensorDesc().getDims() != dst->getTensorDesc().getDims())
IE_THROW() << "Unimplemented blob transformation from different shapes "; IE_THROW() << "Unimplemented blob transformation from different shapes ";

4
src/inference/src/dev/make_tensor.cpp
View File

@ -512,12 +512,12 @@ public:
auto dims = shape; auto dims = shape;
if (shape.size() == 4) { if (shape.size() == 4) {
// NHWC // NHWC
if (shape[1] == shape[2]) { if ((shape[1] == shape[2]) && (shape[1] != shape[3])) {
blk_order = {0, 2, 3, 1}; blk_order = {0, 2, 3, 1};
dims[1] = shape[3]; dims[1] = shape[3];
dims[2] = shape[1]; dims[2] = shape[1];
dims[3] = shape[2]; dims[3] = shape[2];
} }
} }
ie::SizeVector dim_offset(shape.size(), 0); ie::SizeVector dim_offset(shape.size(), 0);
ie::SizeVector blk_strides; ie::SizeVector blk_strides;