Home
last modified time | relevance | path

Searched refs:harness_2d_reduction (Results 1 – 21 of 21) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_primitive_conf.hpp63 harness_2d_reduction, enumerator
H A Djit_sve_512_convolution.cpp1228 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 Djit_sve_512_conv_kernel.cpp3597 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 Djit_primitive_conf.hpp63 harness_2d_reduction, enumerator
H A Djit_sve_512_convolution.cpp1228 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 Djit_sve_512_conv_kernel.cpp3597 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 Djit_avx512_common_convolution.cpp1282 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 Djit_primitive_conf.hpp61 harness_2d_reduction, enumerator
H A Djit_avx512_common_conv_kernel.cpp4391 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 Djit_avx512_core_bf16_conv_kernel.cpp2599 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 Djit_avx512_core_amx_conv_kernel.cpp4218 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 Djit_avx512_core_amx_convolution.cpp1992 case harness_2d_reduction: in execute_backward_weights()
H A Djit_avx512_core_bf16_convolution.cpp1961 case harness_2d_reduction: in execute_backward_weights()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx512_common_convolution.cpp1282 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 Djit_primitive_conf.hpp61 harness_2d_reduction, enumerator
H A Djit_avx512_common_conv_kernel.cpp4379 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 Djit_avx512_core_bf16_conv_kernel.cpp2590 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 Djit_brgemm_inner_product_utils.cpp759 : harness_2d_reduction; in init_ip_conf_bwd_w()
H A Djit_avx512_core_amx_conv_kernel.cpp4244 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 Djit_avx512_core_amx_convolution.cpp1973 case harness_2d_reduction: in execute_backward_weights()
H A Djit_avx512_core_bf16_convolution.cpp1961 case harness_2d_reduction: in execute_backward_weights()