/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_convolution_backward_data_common.hpp | 39 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 D | test_convolution_forward_common.hpp | 44 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 D | test_convolution_eltwise_forward_common.hpp | 41 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 D | test_convolution_backward_weights_common.hpp | 75 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 D | test_inner_product_backward_data.cpp | 48 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 D | test_inner_product_backward_weights.cpp | 66 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 D | test_inner_product_forward.cpp | 49 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 D | test_convolution_backward_data_common.hpp | 39 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 D | test_convolution_forward_common.hpp | 44 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 D | test_convolution_eltwise_forward_common.hpp | 41 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 D | test_convolution_backward_weights_common.hpp | 75 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 D | test_inner_product_backward_data.cpp | 48 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 D | test_inner_product_forward.cpp | 49 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 D | test_inner_product_backward_weights.cpp | 66 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 D | jit_brgemm_conv_utils.cpp | 512 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()
|