Merge pull request #12 from PoliOwl/non_zero_dynamic_shape

[VPU] Fixed allocating dynamic blobs in myriad_infer_request
AlexeyLebedev1 · Aug 26, 2021 · fee51a5 · fee51a5
2 parents 9dd3eb0 + 07301a7
commit fee51a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference-engine/src/vpu/myriad_plugin/myriad_infer_request.cpp b/inference-engine/src/vpu/myriad_plugin/myriad_infer_request.cpp
@@ -266,7 +266,7 @@ void MyriadInferRequest::GetResult() {
                 "Can not find tensor descriptor by plugin for {} output", ieBlobName);
             const auto& dynOutputDesc = descFromPlugin->second;
 
-            if (ieBlob->getTensorDesc().getLayout() != dynOutputDesc.getLayout()) {
+            if (ieBlob->getTensorDesc().getDims() != dynOutputDesc.getDims()) {
                 ieBlob->deallocate();
                 ieBlob->getTensorDesc().reshape(dynOutputDesc.getDims(), dynOutputDesc.getLayout());
                 ieBlob->allocate();