| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_dw_conv_kernel_utils.cpp | 551 jcp.harness = is_data_layout_nxc ? harness_nxc : harness_mb_reduction; in init_conf()
563 if (jcp.harness == harness_mb_reduction) { in init_scratchpad()
608 if (jcp.harness == harness_mb_reduction) { in balance()
|
| H A D | jit_uni_dw_convolution.hpp | 247 case harness_mb_reduction: in execute()
|
| H A D | jit_brgemm_inner_product_utils.cpp | 758 ? harness_mb_reduction in init_ip_conf_bwd_w()
956 && (jbgp.nthr_mb > 1 || jbgp.harness == harness_mb_reduction)) { in init_scratchpad()
|
| H A D | jit_primitive_conf.hpp | 63 harness_mb_reduction, enumerator
|
| H A D | jit_avx512_common_convolution.cpp | 2162 case harness_mb_reduction: in execute_backward_weights()
2189 case harness_mb_reduction: in execute_backward_weights()
2201 case harness_mb_reduction: in execute_backward_weights()
2219 case harness_mb_reduction: { in execute_backward_weights()
|
| H A D | jit_brgemm_inner_product.cpp | 1075 if (jbgp.nthr_mb > 1 || jbgp.harness == harness_mb_reduction) { in get_wei_acc_ptr()
1348 const bool osc_loop_outermost = jbgp.harness == harness_mb_reduction; in compute_diff_weights_and_bias()
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 2594 case harness_mb_reduction: in maybe_compute_diff_bias()
4052 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
4451 : harness_mb_reduction); in init_conf()
4457 case harness_mb_reduction: jcp.nthr_mb_work = jcp.mb; break; in init_conf()
|
| H A D | jit_avx512_common_conv_kernel.cpp | 4193 assert(one_of(jcp.harness, harness_mb_reduction, harness_3d_reduction)); in compute_oh_loop_common()
4642 assert(jcp.harness == harness_mb_reduction); in compute_full_spat_loop()
5180 assert(jcp.harness == harness_mb_reduction); in flat_4ops_compute()
5369 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
5986 : ndims == 5 ? harness_3d_reduction : harness_mb_reduction; in init_conf()
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 4248 case harness_mb_reduction: in maybe_compute_diff_bias()
4814 case harness_mb_reduction: in compute_loop()
5116 : harness_mb_reduction); in init_conf()
5121 case harness_mb_reduction: jcp.nthr_mb_work = jcp.mb; break; in init_conf()
|
| H A D | jit_avx512_core_amx_convolution.cpp | 1984 case harness_mb_reduction: in execute_backward_weights()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_dw_conv_kernel_utils.hpp | 670 jcp.harness = is_data_layout_nxc ? harness_nxc : harness_mb_reduction; in init_conf()
682 if (jcp.harness == harness_mb_reduction) { in init_scratchpad()
727 if (jcp.harness == harness_mb_reduction) { in balance()
|
| H A D | jit_uni_dw_convolution.hpp | 248 case harness_mb_reduction: in execute()
|
| H A D | jit_primitive_conf.hpp | 63 harness_mb_reduction, enumerator
|
| H A D | jit_avx512_common_convolution.cpp | 2160 case harness_mb_reduction: in execute_backward_weights()
2187 case harness_mb_reduction: in execute_backward_weights()
2199 case harness_mb_reduction: in execute_backward_weights()
2217 case harness_mb_reduction: { in execute_backward_weights()
|
| H A D | jit_avx512_core_bf16_conv_kernel.cpp | 2603 case harness_mb_reduction: in maybe_compute_diff_bias()
4061 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
4460 : harness_mb_reduction); in init_conf()
4466 case harness_mb_reduction: jcp.nthr_mb_work = jcp.mb; break; in init_conf()
|
| H A D | jit_avx512_common_conv_kernel.cpp | 4205 assert(one_of(jcp.harness, harness_mb_reduction, harness_3d_reduction)); in compute_oh_loop_common()
4641 assert(jcp.harness == harness_mb_reduction); in compute_full_spat_loop()
5179 assert(jcp.harness == harness_mb_reduction); in flat_4ops_compute()
5368 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
5985 : ndims == 5 ? harness_3d_reduction : harness_mb_reduction; in init_conf()
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 4222 case harness_mb_reduction: in maybe_compute_diff_bias()
4788 case harness_mb_reduction: in compute_loop()
5090 : harness_mb_reduction); in init_conf()
5095 case harness_mb_reduction: jcp.nthr_mb_work = jcp.mb; break; in init_conf()
|
| H A D | jit_avx512_core_amx_convolution.cpp | 2003 case harness_mb_reduction: in execute_backward_weights()
|
| H A D | jit_avx512_core_bf16_convolution.cpp | 1972 case harness_mb_reduction: in execute_backward_weights()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_primitive_conf.hpp | 65 harness_mb_reduction, enumerator
|
| H A D | jit_sve_512_convolution.cpp | 1770 case harness_mb_reduction: in execute_backward_weights()
1788 case harness_mb_reduction: in execute_backward_weights()
1800 case harness_mb_reduction: in execute_backward_weights()
1818 case harness_mb_reduction: { in execute_backward_weights()
|
| H A D | jit_sve_512_conv_kernel.cpp | 3392 assert(one_of(jcp.harness, harness_mb_reduction, harness_3d_reduction)); in compute_oh_loop_common()
3876 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
4161 : ndims == 5 ? harness_3d_reduction : harness_mb_reduction; in init_conf()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_primitive_conf.hpp | 65 harness_mb_reduction, enumerator
|
| H A D | jit_sve_512_convolution.cpp | 1770 case harness_mb_reduction: in execute_backward_weights()
1788 case harness_mb_reduction: in execute_backward_weights()
1800 case harness_mb_reduction: in execute_backward_weights()
1818 case harness_mb_reduction: { in execute_backward_weights()
|
| H A D | jit_sve_512_conv_kernel.cpp | 3392 assert(one_of(jcp.harness, harness_mb_reduction, harness_3d_reduction)); in compute_oh_loop_common()
3876 case harness_mb_reduction: compute_oh_loop_common(); break; in compute_loop()
4161 : ndims == 5 ? harness_3d_reduction : harness_mb_reduction; in init_conf()
|