Home
last modified time | relevance | path

Searched refs:padded_ic (Results 1 – 15 of 15) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/
H A Dtest_convolution_backward_data_common.hpp39 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_conv_bwd_data() local
49 memory::dim sidx = mb * padded_ic * c.ih * c.iw in compute_ref_conv_bwd_data()
50 + g * padded_ic / c.ng * c.ih * c.iw + ic * c.ih * c.iw in compute_ref_conv_bwd_data()
69 * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_data()
70 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_data()
H A Dtest_convolution_forward_common.hpp44 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_fwd() local
65 memory::dim iidx = n * padded_ic * c.ih * c.iw in compute_ref_conv_fwd()
66 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_fwd()
68 memory::dim widx = g * padded_oc / c.ng * padded_ic in compute_ref_conv_fwd()
70 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_fwd()
H A Dtest_convolution_eltwise_forward_common.hpp41 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_eltwise_fwd() local
67 memory::dim iidx = n * padded_ic * c.ih * c.iw in compute_ref_conv_eltwise_fwd()
68 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_eltwise_fwd()
71 + g * padded_oc / c.ng * padded_ic / c.ng * c.kh in compute_ref_conv_eltwise_fwd()
73 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_eltwise_fwd()
H A Dtest_convolution_backward_weights_common.hpp75 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_bwd_weights() local
82 = g * padded_oc / c.ng * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_weights()
83 + oc * padded_ic / c.ng * c.kh * c.kw + ic * c.kh * c.kw in compute_ref_conv_bwd_weights()
101 memory::dim sidx = mb * padded_ic * c.ih * c.iw in compute_ref_conv_bwd_weights()
102 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_bwd_weights()
H A Dtest_inner_product_backward_data.cpp48 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_inner_product_bwd_data() local
55 memory::dim dsidx = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
62 memory::dim widx = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
H A Dtest_inner_product_backward_weights.cpp66 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_inner_product_bwd_weights() local
77 = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_weights()
86 = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_weights()
H A Dtest_inner_product_forward.cpp49 auto padded_ic = src_mdw.padded_dims()[1]; in compute_ref_inner_product_fwd() local
59 memory::dim iidx = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_fwd()
62 memory::dim widx = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_fwd()
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/
H A Dtest_convolution_backward_data_common.hpp39 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_conv_bwd_data() local
49 memory::dim sidx = mb * padded_ic * c.ih * c.iw in compute_ref_conv_bwd_data()
50 + g * padded_ic / c.ng * c.ih * c.iw + ic * c.ih * c.iw in compute_ref_conv_bwd_data()
69 * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_data()
70 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_data()
H A Dtest_convolution_forward_common.hpp44 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_fwd() local
65 memory::dim iidx = n * padded_ic * c.ih * c.iw in compute_ref_conv_fwd()
66 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_fwd()
68 memory::dim widx = g * padded_oc / c.ng * padded_ic in compute_ref_conv_fwd()
70 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_fwd()
H A Dtest_convolution_eltwise_forward_common.hpp41 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_eltwise_fwd() local
67 memory::dim iidx = n * padded_ic * c.ih * c.iw in compute_ref_conv_eltwise_fwd()
68 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_eltwise_fwd()
71 + g * padded_oc / c.ng * padded_ic / c.ng * c.kh in compute_ref_conv_eltwise_fwd()
73 + oc * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_eltwise_fwd()
H A Dtest_convolution_backward_weights_common.hpp75 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_conv_bwd_weights() local
82 = g * padded_oc / c.ng * padded_ic / c.ng * c.kh * c.kw in compute_ref_conv_bwd_weights()
83 + oc * padded_ic / c.ng * c.kh * c.kw + ic * c.kh * c.kw in compute_ref_conv_bwd_weights()
101 memory::dim sidx = mb * padded_ic * c.ih * c.iw in compute_ref_conv_bwd_weights()
102 + g * padded_ic / c.ng * c.ih * c.iw in compute_ref_conv_bwd_weights()
H A Dtest_inner_product_backward_data.cpp48 auto padded_ic = diff_src_d.data.padded_dims[1]; in compute_ref_inner_product_bwd_data() local
55 memory::dim dsidx = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
62 memory::dim widx = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_data()
H A Dtest_inner_product_forward.cpp49 auto padded_ic = src_mdw.padded_dims()[1]; in compute_ref_inner_product_fwd() local
59 memory::dim iidx = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_fwd()
62 memory::dim widx = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_fwd()
H A Dtest_inner_product_backward_weights.cpp66 auto padded_ic = src_d.data.padded_dims[1]; in compute_ref_inner_product_bwd_weights() local
77 = oc * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_weights()
86 = n * padded_ic * ipd.kd * ipd.kh * ipd.kw in compute_ref_inner_product_bwd_weights()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_brgemm_conv_utils.cpp512 const auto padded_ic = last_ic_block_size * (is_ic_padded ? 16 : 1); in select_ic_block() local
566 (exec_type == exec_trans) ? rnd_up(ic, padded_ic) : ic, in select_ic_block()
585 const auto padded_ic = last_ic_block_size * (is_ic_padded ? 16 : 1); in estimate_brgemm_ur() local
587 icp = rnd_up(ic, padded_ic); in estimate_brgemm_ur()