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

[CPU] Fixed strides for Pad after optimized Split (#4556)

Browse Source
This commit is contained in:
Alexandra Sidorova 2021-03-03 09:28:01 +03:00 committed by GitHub
parent a35dfc4c5a
commit ebe63f557c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
inference-engine/src/mkldnn_plugin/nodes/mkldnn_pad_node.cpp
View File

@ -133,8 +133,13 @@ void MKLDNNPadNode::createPrimitive() {
params.srcDims = getParentEdgeAt(0)->getBlob()->getTensorDesc().getBlockingDesc().getBlockDims();
params.dstDims = getChildEdgeAt(0)->getBlob()->getTensorDesc().getBlockingDesc().getBlockDims();
params.srcStrides = getParentEdgeAt(0)->getBlob()->getTensorDesc().getBlockingDesc().getStrides();
params.dstStrides = getChildEdgeAt(0)->getBlob()->getTensorDesc().getBlockingDesc().getStrides();
size_t nDims = params.srcDims.size();
params.srcStrides.resize(nDims, 1);
params.dstStrides.resize(nDims, 1);
for (int i = nDims - 2; i >= 0; i--) {
params.srcStrides[i] = params.srcStrides[i + 1] * params.srcDims[i + 1];
params.dstStrides[i] = params.dstStrides[i + 1] * params.dstDims[i + 1];
}
if (getParentEdgeAt(0)->getMemory().GetDesc().isBlockedCFormat()) {
padsBegin[1] /= params.srcDims[params.srcDims.size() - 1];