| /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()
|