[CPU] Reference node shape infer is now always called (#8517)

This commit is contained in:
Maksim Kutakov authored and GitHub committed 2021-11-14 11:16:36 +03:00
1 parent 57900a4a18
commit 3c985ed592
2 files changed
+5

No files matched your search

@@ -105,3 +105,7 @@ void MKLDNNReferenceNode::executeDynamicImpl(mkldnn::stream strm) {
bool MKLDNNReferenceNode::created() const {
return getType() == Reference;
}
bool MKLDNNReferenceNode::needShapeInfer() const {
return true;
}
@@ -19,6 +19,7 @@ public:
bool created() const override;
std::vector<VectorDims> shapeInfer() const override;
bool needShapeInfer() const override;
bool needPrepareParams() const override { return false; }
void executeDynamicImpl(mkldnn::stream strm) override;