[CPU] fixed strides filling in blocking desc (#2353)
authorAnton Voronov <anton.voronov@intel.com>
Tue, 10 Nov 2020 08:25:12 +0000 (11:25 +0300)
committerGitHub <noreply@github.com>
Tue, 10 Nov 2020 08:25:12 +0000 (11:25 +0300)
inference-engine/src/mkldnn_plugin/mkldnn_memory.cpp

index 72a511a..0f2b2f2 100644 (file)
@@ -1392,7 +1392,7 @@ MKLDNNMemoryDesc::MKLDNNMemoryDesc(const TensorDesc& tDesc):
 
     if (notDefault) {
         for (size_t i = 0; i < strides.size() && i < desc.data.ndims; i++) {
-            desc.data.layout_desc.blocking.strides[0][i] = static_cast<ptrdiff_t>(strides[order[i]]);
+            desc.data.layout_desc.blocking.strides[0][order[i]] = static_cast<ptrdiff_t>(strides[i]);
         }
     }
 }