/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_sse41_1x1_conv_kernel_f32.cpp | 241 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 D | jit_avx2_1x1_convolution.cpp | 185 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 D | jit_avx2_1x1_conv_kernel_f32.cpp | 304 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 D | jit_avx512_common_1x1_convolution.cpp | 177 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 D | jit_sse41_1x1_convolution.cpp | 155 par_conv.first_last_flag = 0 | (icb == 0) * FLAG_REDUCE_FIRST in execute_forward_thr()
|
H A D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 386 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 D | jit_avx512_core_bf16_1x1_convolution.cpp | 217 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 D | jit_primitive_conf.hpp | 77 FLAG_REDUCE_FIRST = 1 << 8, enumerator
|
H A D | jit_brgemm_post_ops.hpp | 103 test(reg_flag, FLAG_REDUCE_FIRST); in loop_by_N()
|
H A D | jit_brgemm_inner_product.cpp | 1225 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 D | jit_sse41_1x1_conv_kernel_f32.cpp | 234 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 D | jit_avx2_1x1_convolution.cpp | 185 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 D | jit_avx2_1x1_conv_kernel_f32.cpp | 295 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 D | jit_avx512_common_1x1_convolution.cpp | 177 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 D | jit_sse41_1x1_convolution.cpp | 155 par_conv.first_last_flag = 0 | (icb == 0) * FLAG_REDUCE_FIRST in execute_forward_thr()
|
H A D | jit_avx512_core_bf16_1x1_conv_kernel.cpp | 367 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 D | jit_avx512_core_bf16_1x1_convolution.cpp | 217 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 D | jit_primitive_conf.hpp | 77 FLAG_REDUCE_FIRST = 1 << 8, enumerator
|
H A D | jit_brgemm_post_ops.hpp | 103 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 D | jit_primitive_conf.hpp | 79 FLAG_REDUCE_FIRST = 1 << 8, enumerator
|
H A D | jit_sve_512_1x1_convolution.cpp | 149 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 D | jit_sve_512_1x1_conv_kernel.cpp | 349 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 D | jit_primitive_conf.hpp | 79 FLAG_REDUCE_FIRST = 1 << 8, enumerator
|
H A D | jit_sve_512_1x1_convolution.cpp | 149 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 D | jit_sve_512_1x1_conv_kernel.cpp | 349 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop() 371 tst(reg_reduce_pos_flag, FLAG_REDUCE_FIRST); in reduce_loop()
|