1 files changed, 3 insertions, 0 deletions
diff --git a/inference-engine/thirdparty/mkl-dnn/src/common/batch_normalization_pd.hpp b/inference-engine/thirdparty/mkl-dnn/src/common/batch_normalization_pd.hpp
index 4f7ae928a..96f9cf90f 100644
--- a/inference-engine/thirdparty/mkl-dnn/src/common/batch_normalization_pd.hpp
+++ b/inference-engine/thirdparty/mkl-dnn/src/common/batch_normalization_pd.hpp
@@ -96,6 +96,9 @@ struct batch_normalization_pd_t: public primitive_desc_t {
 
     inline int ndims() const { return desc_.data_desc.ndims; }
 
+    bool has_zero_dim_memory() const
+    { return memory_desc_wrapper(desc_.data_desc).has_zero_dim(); }
+
 protected:
     batch_normalization_desc_t desc_;
     const batch_normalization_fwd_pd_t *hint_fwd_pd_;