diff options
author | Alexey Suhov <asuhov@users.noreply.github.com> | 2018-11-23 16:19:43 +0300 |
---|---|---|
committer | openvino-pushbot <44090433+openvino-pushbot@users.noreply.github.com> | 2018-11-23 16:19:43 +0300 |
commit | 55a41d7570f78aaea0d6764d157dd7434730d56f (patch) | |
tree | ba022c71609b93d51119bcb25e5ccb8c7147dbd3 /inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp | |
parent | 54eab180361ec09fbd82e2bb62adfeb521275774 (diff) | |
download | dldt-55a41d7570f78aaea0d6764d157dd7434730d56f.tar.gz dldt-55a41d7570f78aaea0d6764d157dd7434730d56f.tar.bz2 dldt-55a41d7570f78aaea0d6764d157dd7434730d56f.zip |
Publishing R4 (#41)
* Publishing R4
Diffstat (limited to 'inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp')
-rw-r--r-- | inference-engine/thirdparty/mkl-dnn/src/common/pooling_pd.hpp | 7 |
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_; |