/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 63 harness_2d_reduction, enumerator
|
H A D | jit_sve_512_convolution.cpp | 1228 int oh_reduce = jcp.harness == harness_2d_reduction ? jcp.oh : 1; in thread_info_t() 1760 case harness_2d_reduction: in execute_backward_weights() 1782 case harness_2d_reduction: in execute_backward_weights() 1801 case harness_2d_reduction: in execute_backward_weights() 1813 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_sve_512_conv_kernel.cpp | 3597 assert(jcp.harness == harness_2d_reduction); in compute_oh_loop_partial() 3874 case harness_2d_reduction: compute_oh_loop_partial(); break; in compute_loop() 4164 jcp.harness = harness_2d_reduction; // 2d harness with oh reduction in init_conf() 4286 int ih_reduce = j.harness == harness_2d_reduction ? ih : 1; in balance() 4287 int oh_reduce = j.harness == harness_2d_reduction ? oh : 1; in balance() 4288 int ih_no_reduce = j.harness == harness_2d_reduction ? 1 : ih; in balance() 4289 int oh_no_reduce = j.harness == harness_2d_reduction ? 1 : oh; in balance()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 63 harness_2d_reduction, enumerator
|
H A D | jit_sve_512_convolution.cpp | 1228 int oh_reduce = jcp.harness == harness_2d_reduction ? jcp.oh : 1; in thread_info_t() 1760 case harness_2d_reduction: in execute_backward_weights() 1782 case harness_2d_reduction: in execute_backward_weights() 1801 case harness_2d_reduction: in execute_backward_weights() 1813 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_sve_512_conv_kernel.cpp | 3597 assert(jcp.harness == harness_2d_reduction); in compute_oh_loop_partial() 3874 case harness_2d_reduction: compute_oh_loop_partial(); break; in compute_loop() 4164 jcp.harness = harness_2d_reduction; // 2d harness with oh reduction in init_conf() 4286 int ih_reduce = j.harness == harness_2d_reduction ? ih : 1; in balance() 4287 int oh_reduce = j.harness == harness_2d_reduction ? oh : 1; in balance() 4288 int ih_no_reduce = j.harness == harness_2d_reduction ? 1 : ih; in balance() 4289 int oh_no_reduce = j.harness == harness_2d_reduction ? 1 : oh; in balance()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx512_common_convolution.cpp | 1282 int oh_reduce = jcp.harness == harness_2d_reduction ? jcp.oh : 1; in thread_info_t() 2150 case harness_2d_reduction: in execute_backward_weights() 2181 case harness_2d_reduction: in execute_backward_weights() 2200 case harness_2d_reduction: in execute_backward_weights() 2212 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_primitive_conf.hpp | 61 harness_2d_reduction, enumerator
|
H A D | jit_avx512_common_conv_kernel.cpp | 4391 assert(jcp.harness == harness_2d_reduction); in compute_oh_loop_partial() 5366 case harness_2d_reduction: compute_oh_loop_partial(); break; in compute_loop() 5988 jcp.harness = harness_2d_reduction; // 2d harness with oh reduction in init_conf() 6149 int ih_reduce = j.harness == harness_2d_reduction ? ih : 1; in balance() 6150 int oh_reduce = j.harness == harness_2d_reduction ? oh : 1; in balance() 6151 int ih_no_reduce = j.harness == harness_2d_reduction ? 1 : ih; in balance() 6152 int oh_no_reduce = j.harness == harness_2d_reduction ? 1 : oh; in balance()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 2599 case harness_2d_reduction: in maybe_compute_diff_bias() 4060 case harness_2d_reduction: compute_oh_loop_common(true); break; in compute_loop() 4459 : (ndims == 4) ? harness_2d_reduction in init_conf() 4463 case harness_2d_reduction: jcp.nthr_mb_work = jcp.mb * jcp.oh; break; in init_conf()
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 4218 case harness_2d_reduction: in maybe_compute_diff_bias() 4785 case harness_2d_reduction: in compute_loop() 5089 : (ndims == 4) ? harness_2d_reduction in init_conf() 5092 case harness_2d_reduction: jcp.nthr_mb_work = jcp.mb * jcp.oh; break; in init_conf()
|
H A D | jit_avx512_core_amx_convolution.cpp | 1992 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1961 case harness_2d_reduction: in execute_backward_weights()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_avx512_common_convolution.cpp | 1282 int oh_reduce = jcp.harness == harness_2d_reduction ? jcp.oh : 1; in thread_info_t() 2152 case harness_2d_reduction: in execute_backward_weights() 2183 case harness_2d_reduction: in execute_backward_weights() 2202 case harness_2d_reduction: in execute_backward_weights() 2214 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_primitive_conf.hpp | 61 harness_2d_reduction, enumerator
|
H A D | jit_avx512_common_conv_kernel.cpp | 4379 assert(jcp.harness == harness_2d_reduction); in compute_oh_loop_partial() 5367 case harness_2d_reduction: compute_oh_loop_partial(); break; in compute_loop() 5989 jcp.harness = harness_2d_reduction; // 2d harness with oh reduction in init_conf() 6150 int ih_reduce = j.harness == harness_2d_reduction ? ih : 1; in balance() 6151 int oh_reduce = j.harness == harness_2d_reduction ? oh : 1; in balance() 6152 int ih_no_reduce = j.harness == harness_2d_reduction ? 1 : ih; in balance() 6153 int oh_no_reduce = j.harness == harness_2d_reduction ? 1 : oh; in balance()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 2590 case harness_2d_reduction: in maybe_compute_diff_bias() 4051 case harness_2d_reduction: compute_oh_loop_common(true); break; in compute_loop() 4450 : (ndims == 4) ? harness_2d_reduction in init_conf() 4454 case harness_2d_reduction: jcp.nthr_mb_work = jcp.mb * jcp.oh; break; in init_conf()
|
H A D | jit_brgemm_inner_product_utils.cpp | 759 : harness_2d_reduction; in init_ip_conf_bwd_w()
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 4244 case harness_2d_reduction: in maybe_compute_diff_bias() 4811 case harness_2d_reduction: in compute_loop() 5115 : (ndims == 4) ? harness_2d_reduction in init_conf() 5118 case harness_2d_reduction: jcp.nthr_mb_work = jcp.mb * jcp.oh; break; in init_conf()
|
H A D | jit_avx512_core_amx_convolution.cpp | 1973 case harness_2d_reduction: in execute_backward_weights()
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1961 case harness_2d_reduction: in execute_backward_weights()
|