File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments