Home
last modified time | relevance | path

Searched refs:nchw_pooling_bwd_t (Results 1 – 6 of 6) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dnchw_pooling.hpp100 struct nchw_pooling_bwd_t : public primitive_t { struct
104 DECLARE_COMMON_PD_T("simple_nchw:any", nchw_pooling_bwd_t);
183 nchw_pooling_bwd_t(const pd_t *apd) : primitive_t(apd) {} in nchw_pooling_bwd_t() function
H A Dcpu_pooling_list.cpp59 CPU_INSTANCE(nchw_pooling_bwd_t<bf16>)
61 CPU_INSTANCE(nchw_pooling_bwd_t<f32>)
H A Dnchw_pooling.cpp335 status_t nchw_pooling_bwd_t<d_type>::execute_backward( in execute_backward()
481 status_t nchw_pooling_bwd_t<data_type::bf16>::execute_backward( in execute_backward()
681 template struct nchw_pooling_bwd_t<data_type::f32>; variable
683 template struct nchw_pooling_bwd_t<data_type::bf16>; variable
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dnchw_pooling.hpp100 struct nchw_pooling_bwd_t : public primitive_t { struct
104 DECLARE_COMMON_PD_T("simple_nchw:any", nchw_pooling_bwd_t);
183 nchw_pooling_bwd_t(const pd_t *apd) : primitive_t(apd) {} in nchw_pooling_bwd_t() argument
H A Dcpu_pooling_list.cpp77 CPU_INSTANCE(nchw_pooling_bwd_t<bf16>)
78 CPU_INSTANCE(nchw_pooling_bwd_t<f32>)
H A Dnchw_pooling.cpp341 status_t nchw_pooling_bwd_t<d_type>::execute_backward( in execute_backward()
489 status_t nchw_pooling_bwd_t<data_type::bf16>::execute_backward( in execute_backward()
691 template struct nchw_pooling_bwd_t<data_type::f32>; variable
693 template struct nchw_pooling_bwd_t<data_type::bf16>; variable