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

[CPU] Disabled inplace for const input in Scatter node (#4148)

Browse Source
This commit is contained in:
Yury Gaydaychuk authored and GitHub committed 2021-02-04 22:42:58 +03:00
1 parent 1bd7b37419
commit bf05b4d4f6
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
inference-engine/src/mkldnn_plugin/nodes/mkldnn_scatter_update_node.cpp
+2 -1
View File
@@ -167,7 +167,8 @@ void MKLDNNScatterUpdateNode::initSupportedPrimitiveDescriptors() {
auto dataType = MKLDNNExtensionUtils::IEPrecisionToDataType(dataPrec);
dataSize = MKLDNNExtensionUtils::sizeOfDataType(dataType);
bool canBeInplace = getParentEdgeAt(DATA_ID)->getParent()->getChildEdges().size() == 1;
bool canBeInplace = getParentEdgeAt(DATA_ID)->getParent()->getChildEdges().size() == 1 &&
!getParentEdgeAt(DATA_ID)->getParent()->isConstant();
InferenceEngine::LayerConfig config;
config.dynBatchSupport = false;