/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 239 p.first_last_flag |= FLAG_OC_LAST; in execute_forward_thr() 241 p.first_last_flag &= ~FLAG_OC_LAST; in execute_forward_thr()
|
H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 240 p.first_last_flag |= FLAG_OC_LAST; in execute_forward_thr() 242 p.first_last_flag &= ~FLAG_OC_LAST; in execute_forward_thr()
|
H A D | jit_avx2_convolution.cpp | 152 if (ocbb == ocb_work - 1) par_conv.oc_flag |= FLAG_OC_LAST; in execute_forward() 450 par_conv.flags |= FLAG_OC_LAST; in execute_backward_weights()
|
H A D | jit_avx2_conv_kernel_f32.cpp | 244 test(reg_oc_flag, FLAG_OC_LAST); in apply_postops() 340 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 458 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 479 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 1543 test(reg_ci_flag, FLAG_OC_LAST); in compute_ic_block_step()
|
H A D | jit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp | 265 test(reg_reduce_pos_flag, FLAG_OC_LAST); in apply_postops() 543 test(reg_reduce_pos_flag, FLAG_OC_LAST); in reduce_loop() 632 test(reg_reduce_pos_flag, FLAG_OC_LAST); in generate()
|
H A D | jit_uni_dw_conv_kernel_f32.cpp | 1561 and_(reg_exec_flags, FLAG_OC_LAST); in compute_single_ch_block_bias() 1606 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop_bias() 1695 and_(reg_exec_flags, FLAG_OC_LAST); in zero_filter_ch_loop() 1794 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop()
|
H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 1110 and_(reg_exec_flags, FLAG_OC_LAST); in compute_single_ch_block_bias() 1145 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop_bias() 1283 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop()
|
H A D | jit_uni_dw_convolution.cpp | 390 = (g + g_step) >= jcp.nb_ch ? FLAG_OC_LAST : 0; in execute_backward_weights_nxc() 535 unsigned char last_g_flag = g == nb_ch - 1 ? FLAG_OC_LAST : 0; in execute_backward_weights()
|
H A D | jit_primitive_conf.hpp | 72 FLAG_OC_LAST = 1 << 3, enumerator
|
H A D | jit_uni_x8s8s32x_1x1_conv_kernel.cpp | 474 test(reg_reduce_pos_flag, FLAG_OC_LAST); in reduce_loop()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 240 p.first_last_flag |= FLAG_OC_LAST; in execute_forward_thr() 242 p.first_last_flag &= ~FLAG_OC_LAST; in execute_forward_thr()
|
H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 239 p.first_last_flag |= FLAG_OC_LAST; in execute_forward_thr() 241 p.first_last_flag &= ~FLAG_OC_LAST; in execute_forward_thr()
|
H A D | jit_avx2_convolution.cpp | 152 if (ocbb == ocb_work - 1) par_conv.oc_flag |= FLAG_OC_LAST; in execute_forward() 450 par_conv.flags |= FLAG_OC_LAST; in execute_backward_weights()
|
H A D | jit_avx2_conv_kernel_f32.cpp | 233 test(reg_oc_flag, FLAG_OC_LAST); in apply_postops() 329 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 447 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 468 test(reg_oc_flag, FLAG_OC_LAST); in width_blk_step() 1533 test(reg_ci_flag, FLAG_OC_LAST); in compute_ic_block_step()
|
H A D | jit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp | 257 test(reg_reduce_pos_flag, FLAG_OC_LAST); in apply_postops() 548 test(reg_reduce_pos_flag, FLAG_OC_LAST); in reduce_loop() 637 test(reg_reduce_pos_flag, FLAG_OC_LAST); in generate()
|
H A D | jit_uni_dw_conv_kernel_f32.cpp | 1534 and_(reg_exec_flags, FLAG_OC_LAST); in compute_single_ch_block_bias() 1579 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop_bias() 1668 and_(reg_exec_flags, FLAG_OC_LAST); in zero_filter_ch_loop() 1767 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop()
|
H A D | jit_avx512_core_bf16_dw_conv_kernel.cpp | 1105 and_(reg_exec_flags, FLAG_OC_LAST); in compute_single_ch_block_bias() 1140 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop_bias() 1278 and_(reg_exec_flags, FLAG_OC_LAST); in compute_ch_loop()
|
H A D | jit_uni_dw_convolution.cpp | 392 = (g + g_step) >= jcp.nb_ch ? FLAG_OC_LAST : 0; in execute_backward_weights_nxc() 535 unsigned char last_g_flag = g == nb_ch - 1 ? FLAG_OC_LAST : 0; in execute_backward_weights()
|
H A D | jit_primitive_conf.hpp | 72 FLAG_OC_LAST = 1 << 3, enumerator
|
H A D | jit_uni_x8s8s32x_1x1_conv_kernel.cpp | 480 test(reg_reduce_pos_flag, FLAG_OC_LAST); in reduce_loop()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 74 FLAG_OC_LAST = 1 << 3, enumerator
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 74 FLAG_OC_LAST = 1 << 3, enumerator
|