/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
H A D | xe_hpc_convolution.cpp | 89 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 95 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 97 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 267 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 273 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 275 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hp_convolution.cpp | 132 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 138 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 140 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 342 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 348 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 350 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hpc_1x1_convolution.cpp | 37 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 125 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 127 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hp_1x1_convolution.cpp | 37 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 159 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 161 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_lp_x8s8x_1x1_convolution.cpp | 30 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 138 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 140 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_lp_x8s8x_convolution.cpp | 72 const memory_desc_wrapper weights_mdw(wei); in init_conf() local 360 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 362 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 588 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 682 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 684 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hp_bf16_convolution.cpp | 58 const memory_desc_wrapper weights_mdw(diff_weights_md()); in init_conf() local 64 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 66 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hpc_bf16_convolution.cpp | 52 const memory_desc_wrapper weights_mdw(diff_weights_md()); in init_conf() local 58 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 60 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_hp_1st_bwd_convolution.cpp | 43 const memory_desc_wrapper weights_mdw(diff_weights_md()); in init_conf() local 136 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 139 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | gen9_convolution.cpp | 98 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 345 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 348 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 488 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 667 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 670 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 785 const memory_desc_wrapper weights_mdw(diff_weights_md()); in init_conf() local 952 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 955 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | gen9_wino_convolution.cpp | 127 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 258 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 261 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_layer_normalization.cpp | 310 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_lnorm_fwd() local 324 const size_t bias_off = use_weights_bias && !weights_mdw.has_zero_dim() in check_lnorm_fwd() 325 ? weights_mdw.off_l(C, true) in check_lnorm_fwd() 395 ? weights_data[weights_mdw.off_l(c, true)] in check_lnorm_fwd() 398 ? bias_data[weights_mdw.off_l(c, true)] in check_lnorm_fwd() 446 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_lnorm_bwd() local 560 ? weights_data[weights_mdw.off_l(c)] in check_lnorm_bwd() 574 ? weights_data[weights_mdw.off_l(c)] in check_lnorm_bwd()
|
H A D | test_batch_normalization_common.hpp | 416 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_bnorm_fwd() local 418 const size_t bias_off = use_weights_bias && !weights_mdw.has_zero_dim() in check_bnorm_fwd() 419 ? weights_mdw.off_l(bp.c, true) in check_bnorm_fwd() 499 ? weights_data[weights_mdw.off_l(c, true)] in check_bnorm_fwd() 502 ? bias_data[weights_mdw.off_l(c, true)] in check_bnorm_fwd() 552 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_bnorm_bwd() local 618 ? weights_data[weights_mdw.off_l(c, true)] in check_bnorm_bwd()
|
H A D | test_inner_product_backward_data.cpp | 43 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in compute_ref_inner_product_bwd_data() local 66 * weights_data[weights_mdw.off_l(widx, true)]; in compute_ref_inner_product_bwd_data() 77 * weights_data[weights_mdw.off_l(widx, true)]; in compute_ref_inner_product_bwd_data()
|
H A D | test_convolution_backward_data_common.hpp | 43 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in compute_ref_conv_bwd_data() local 76 * weights_data[weights_mdw.off_l( in compute_ref_conv_bwd_data()
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_layer_normalization.cpp | 310 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_lnorm_fwd() local 324 const size_t bias_off = use_weights_bias && !weights_mdw.has_zero_dim() in check_lnorm_fwd() 325 ? weights_mdw.off_l(C, true) in check_lnorm_fwd() 395 ? weights_data[weights_mdw.off_l(c, true)] in check_lnorm_fwd() 398 ? bias_data[weights_mdw.off_l(c, true)] in check_lnorm_fwd() 446 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_lnorm_bwd() local 560 ? weights_data[weights_mdw.off_l(c)] in check_lnorm_bwd() 574 ? weights_data[weights_mdw.off_l(c)] in check_lnorm_bwd()
|
H A D | test_batch_normalization_common.hpp | 416 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_bnorm_fwd() local 418 const size_t bias_off = use_weights_bias && !weights_mdw.has_zero_dim() in check_bnorm_fwd() 419 ? weights_mdw.off_l(bp.c, true) in check_bnorm_fwd() 499 ? weights_data[weights_mdw.off_l(c, true)] in check_bnorm_fwd() 502 ? bias_data[weights_mdw.off_l(c, true)] in check_bnorm_fwd() 552 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in check_bnorm_bwd() local 618 ? weights_data[weights_mdw.off_l(c, true)] in check_bnorm_bwd()
|
H A D | test_inner_product_backward_data.cpp | 43 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in compute_ref_inner_product_bwd_data() local 66 * weights_data[weights_mdw.off_l(widx, true)]; in compute_ref_inner_product_bwd_data() 77 * weights_data[weights_mdw.off_l(widx, true)]; in compute_ref_inner_product_bwd_data()
|
H A D | test_convolution_backward_data_common.hpp | 43 const dnnl::impl::memory_desc_wrapper weights_mdw(weights_d.data); in compute_ref_conv_bwd_data() local 76 * weights_data[weights_mdw.off_l( in compute_ref_conv_bwd_data()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
H A D | xe_lp_x8s8x_1x1_convolution.cpp | 30 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 137 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 139 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | xe_lp_x8s8x_convolution.cpp | 72 const memory_desc_wrapper weights_mdw(wei); in init_conf() local 356 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 358 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 582 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 676 conf.wei_tag = weights_mdw.format_kind() == format_kind::any in init_conf() 678 : weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | gen9_convolution.cpp | 96 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 307 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 310 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 432 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 602 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 605 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf() 845 const memory_desc_wrapper weights_mdw(diff_weights_md()); in init_conf() local 1002 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 1005 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
H A D | gen9_wino_convolution.cpp | 127 const memory_desc_wrapper weights_mdw(weights_md()); in init_conf() local 258 if (weights_mdw.format_kind() == format_kind::any) { in init_conf() 261 conf.wei_tag = weights_mdw.matches_one_of_tag(wei_tag); in init_conf()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
H A D | primitive_conf.hpp | 742 const memory_desc_wrapper weights_mdw(&weights_md); in set_default_conf() local 746 const bool with_groups = weights_mdw.ndims() == src_mdw.ndims() + 1; in set_default_conf() 753 conf.ngroups = with_groups ? weights_mdw.dims()[0] : 1; in set_default_conf() 763 conf.kd = (ndims == 5) ? weights_mdw.dims()[with_groups + 2] : 1; in set_default_conf() 764 conf.kh = (ndims == 3) ? 1 : weights_mdw.dims()[with_groups + ndims - 2]; in set_default_conf() 765 conf.kw = weights_mdw.dims()[with_groups + ndims - 1]; in set_default_conf() 788 conf.weights_data_type = weights_mdw.data_type(); in set_default_conf()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
H A D | primitive_conf.hpp | 821 const memory_desc_wrapper weights_mdw(&weights_md); in set_default_conf() local 825 const bool with_groups = weights_mdw.ndims() == src_mdw.ndims() + 1; in set_default_conf() 832 conf.ngroups = with_groups ? weights_mdw.dims()[0] : 1; in set_default_conf() 842 conf.kd = (ndims == 5) ? weights_mdw.dims()[with_groups + 2] : 1; in set_default_conf() 843 conf.kh = (ndims == 3) ? 1 : weights_mdw.dims()[with_groups + ndims - 2]; in set_default_conf() 844 conf.kw = weights_mdw.dims()[with_groups + ndims - 1]; in set_default_conf() 867 conf.weights_data_type = weights_mdw.data_type(); in set_default_conf()
|