Skip to content

Commit 849c55a

Browse files
alexey-varyzginazhai219
authored andcommitted
[FIX] nchw_pooling dense fix
1 parent 710d159 commit 849c55a

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/cpu/nchw_pooling.hpp

+3
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,9 @@ struct nchw_pooling_fwd_t : public primitive_t {
5151
alg_kind::pooling_avg_include_padding,
5252
alg_kind::pooling_avg_exclude_padding),
5353
VERBOSE_BAD_ALGORITHM);
54+
VDISPATCH_POOLING(
55+
memory_desc_wrapper(dst_md()).is_dense(false),
56+
VERBOSE_UNSUPPORTED_SPARSE_CFG);
5457
VDISPATCH_POOLING(utils::everyone_is(d_type, src_md()->data_type,
5558
dst_md()->data_type),
5659
VERBOSE_UNSUPPORTED_DT);

0 commit comments

Comments
 (0)