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

[CPU] fixed strides filling in blocking desc (#2353)

Browse Source
This commit is contained in:
Anton Voronov 2020-11-10 11:25:12 +03:00 committed by GitHub
parent 753d376062
commit be065c0144
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/mkldnn_plugin/mkldnn_memory.cpp
View 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]);
}
}
}