1 files changed, 7 insertions, 0 deletions
diff --git a/inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp b/inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp
index 7ac56973b..ee5f2ae6a 100644
--- a/inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp
+++ b/inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp
@@ -103,6 +103,10 @@ struct pooling_fwd_pd_t: public primitive_desc_t {
         ? desc_.padding[0][2] : desc_.padding[0][1]; }
     inline int padR() const { return is_3d()
         ? desc_.padding[1][2] : desc_.padding[1][1]; }
+
+    bool has_zero_dim_memory() const
+    { return memory_desc_wrapper(desc_.src_desc).has_zero_dim(); }
+
 protected:
     pooling_desc_t desc_;
     const pooling_fwd_pd_t *hint_fwd_pd_;
@@ -187,6 +191,9 @@ struct pooling_bwd_pd_t: public primitive_desc_t {
     inline int padR() const { return is_3d()
         ? desc_.padding[1][2] : desc_.padding[1][1]; }
 
+    bool has_zero_dim_memory() const
+    { return memory_desc_wrapper(desc_.diff_src_desc).has_zero_dim(); }
+
 protected:
     pooling_desc_t desc_;
     const pooling_fwd_pd_t *hint_fwd_pd_;