[CPU] Reference node shape infer is now always called (#8517)
This commit is contained in:
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;
|
||||
|
||||
|
||||
Reference in new issue
Block a user