Home
last modified time | relevance | path

Searched refs:FLAG_REDUCE_FIRST (Results 1 – 25 of 28) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_sse41_1x1_conv_kernel_f32.cpp241 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
277 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
388 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_diff_bias_loop()
H A Djit_avx2_1x1_convolution.cpp185 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
471 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
612 ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_avx2_1x1_conv_kernel_f32.cpp304 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
365 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
541 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_diff_bias_loop()
H A Djit_avx512_common_1x1_convolution.cpp177 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
575 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
914 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_sse41_1x1_convolution.cpp155 par_conv.first_last_flag = 0 | (icb == 0) * FLAG_REDUCE_FIRST in execute_forward_thr()
H A Djit_avx512_core_bf16_1x1_conv_kernel.cpp386 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
390 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
404 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
981 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in compute_diff_bias()
H A Djit_avx512_core_bf16_1x1_convolution.cpp217 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
543 p.first_last_flag = 0 | (ocb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_backward_data_thr()
856 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_primitive_conf.hpp77 FLAG_REDUCE_FIRST = 1 << 8, enumerator
H A Djit_brgemm_post_ops.hpp103 test(reg_flag, FLAG_REDUCE_FIRST); in loop_by_N()
H A Djit_brgemm_inner_product.cpp1225 p.flags = 0 | (is_first ? FLAG_REDUCE_FIRST : 0) in compute_diff_weights_and_bias()
1277 p.flags = 0 | (is_first ? FLAG_REDUCE_FIRST : 0) in compute_diff_weights_and_bias()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_sse41_1x1_conv_kernel_f32.cpp234 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
270 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
381 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_diff_bias_loop()
H A Djit_avx2_1x1_convolution.cpp185 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
471 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
612 ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_avx2_1x1_conv_kernel_f32.cpp295 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
356 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_reduce_loop()
532 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in generate_diff_bias_loop()
H A Djit_avx512_common_1x1_convolution.cpp177 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
576 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
916 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_sse41_1x1_convolution.cpp155 par_conv.first_last_flag = 0 | (icb == 0) * FLAG_REDUCE_FIRST in execute_forward_thr()
H A Djit_avx512_core_bf16_1x1_conv_kernel.cpp367 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
371 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
385 FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in reduce_loop()
962 test(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); // If FLAG_REDUCE_FIRST in compute_diff_bias()
H A Djit_avx512_core_bf16_1x1_convolution.cpp217 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
545 p.first_last_flag = 0 | (ocb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_backward_data_thr()
859 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_primitive_conf.hpp77 FLAG_REDUCE_FIRST = 1 << 8, enumerator
H A Djit_brgemm_post_ops.hpp103 test(reg_flag, FLAG_REDUCE_FIRST); in loop_by_N()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_primitive_conf.hpp79 FLAG_REDUCE_FIRST = 1 << 8, enumerator
H A Djit_sve_512_1x1_convolution.cpp149 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
425 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
663 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_sve_512_1x1_conv_kernel.cpp349 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop()
371 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_primitive_conf.hpp79 FLAG_REDUCE_FIRST = 1 << 8, enumerator
H A Djit_sve_512_1x1_convolution.cpp149 p.first_last_flag = 0 | (icb == 0 ? FLAG_REDUCE_FIRST : 0) in execute_forward_thr()
425 p.first_last_flag = ocb == 0 ? FLAG_REDUCE_FIRST : 0; in execute_backward_data()
663 | (mb_sp_b == mb_sp_b_start ? FLAG_REDUCE_FIRST in execute_backward_weights()
H A Djit_sve_512_1x1_conv_kernel.cpp349 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop()
371 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop()

12