Home
last modified time | relevance | path

Searched refs:FLAG_OC_LAST (Results 1 – 22 of 22) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_core_x8s8s32x_1x1_convolution.cpp239 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 Djit_uni_x8s8s32x_1x1_convolution.cpp240 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 Djit_avx2_convolution.cpp152 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 Djit_avx2_conv_kernel_f32.cpp244 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 Djit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp265 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 Djit_uni_dw_conv_kernel_f32.cpp1561 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 Djit_avx512_core_bf16_dw_conv_kernel.cpp1110 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 Djit_uni_dw_convolution.cpp390 = (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 Djit_primitive_conf.hpp72 FLAG_OC_LAST = 1 << 3, enumerator
H A Djit_uni_x8s8s32x_1x1_conv_kernel.cpp474 test(reg_reduce_pos_flag, FLAG_OC_LAST); in reduce_loop()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_x8s8s32x_1x1_convolution.cpp240 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 Djit_avx512_core_x8s8s32x_1x1_convolution.cpp239 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 Djit_avx2_convolution.cpp152 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 Djit_avx2_conv_kernel_f32.cpp233 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 Djit_avx512_core_x8s8s32x_1x1_conv_kernel.cpp257 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 Djit_uni_dw_conv_kernel_f32.cpp1534 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 Djit_avx512_core_bf16_dw_conv_kernel.cpp1105 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 Djit_uni_dw_convolution.cpp392 = (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 Djit_primitive_conf.hpp72 FLAG_OC_LAST = 1 << 3, enumerator
H A Djit_uni_x8s8s32x_1x1_conv_kernel.cpp480 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 Djit_primitive_conf.hpp74 FLAG_OC_LAST = 1 << 3, enumerator
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_primitive_conf.hpp74 FLAG_OC_LAST = 1 << 3, enumerator