Skip to content

Commit 7fc1e7a

Browse files
author
Roman Dubtsov
committed
fixup: cpu: ref: pooling: dispatch hcdhw to nchw pooling
1 parent a5e6fc5 commit 7fc1e7a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/cpu/nchw_pooling.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@ struct nchw_pooling_bwd_t: public cpu_primitive_t {
111111
diff_dst_pd()->desc()->data_type,
112112
diff_src_pd()->desc()->data_type)
113113
&& utils::one_of(diff_dst_format, nchw, ncdhw)
114+
&& diff_dst_format == nchw
114115
&& (diff_dst_format == diff_src_pd()->desc()->format)
115116
&& attr()->has_default_values();
116117
if (!ok) return status::unimplemented;
@@ -119,7 +120,8 @@ struct nchw_pooling_bwd_t: public cpu_primitive_t {
119120
bool ws_ok = true
120121
&& hint_fwd_pd_
121122
&& hint_fwd_pd_->workspace_pd()
122-
&& utils::one_of(hint_fwd_pd_->workspace_pd()->desc()->format,
123+
&& utils::one_of(
124+
hint_fwd_pd_->workspace_pd()->desc()->format,
123125
nchw, nChw8c, nChw16c);
124126
if (!ws_ok) return status::unimplemented;
125127

0 commit comments

Comments
 (0)