| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_brgemm_1x1_conv.cpp | 161 const auto last_ic_block in init() local
166 ? (dim_t)rnd_up(jcp.ic, last_ic_block) * jcp.oc in init()
167 : (dim_t)rnd_up(jcp.ic, last_ic_block) * jcp.oc_block; in init()
168 wei_ocb_sz = jcp.wei_plain ? jcp.oc_block * last_ic_block in init()
|
| H A D | jit_brgemm_transpose_utils.hpp | 180 size_t last_ic_block; member
|
| H A D | jit_brgemm_conv.hpp | 185 last_ic_block, wei_ic_sz, wei_kw_sz, wei_kh_sz, wei_kd_sz, member
|
| H A D | jit_uni_x8s8s32x_deconvolution.hpp | 67 last_ic_block = 0x2U, enumerator
|
| H A D | jit_uni_x8s8s32x_conv_kernel.hpp | 57 last_ic_block, enumerator
|
| H A D | jit_avx512_core_x8s8s32x_conv_kernel.hpp | 59 last_ic_block, enumerator
|
| H A D | jit_avx512_core_x8s8s32x_deconvolution.hpp | 41 last_ic_block = 0x2U, enumerator
|
| H A D | jit_primitive_conf.hpp | 487 size_t last_ic_block; member
|
| H A D | jit_avx512_core_amx_convolution.cpp | 1273 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights_2d()
1511 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights_3d()
1759 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights()
1794 p.last_ic_block = ((ic_b + 1) >= jcp.nb_ic) ? 1 : 0; in reduce_and_convert_diff_weights_and_bias()
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 52 typedef enum { no_last_block, last_ic_block } ic_block_t; enumerator
|
| H A D | jit_brgemm_conv.cpp | 340 last_ic_block = (wei_type == f32) ? 1 : ((wei_type == bf16) ? 2 : 4); in init()
341 wei_ic_sz = (dim_t)rnd_up(jcp.ic, last_ic_block) * jcp.oc_block; in init()
|
| H A D | jit_brgemm_conv_utils.cpp | 518 const auto last_ic_block in get_brgemm_ur() local
520 brg_strides.stride_b = rnd_up(ic, last_ic_block) in get_brgemm_ur()
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 84 const bool masked_write = ic_tail && last_ic_block_flag == last_ic_block; in compute_ker()
88 const int icb = (last_ic_block_flag == last_ic_block) in compute_ker()
120 if (jcp.is_relo && last_ic_block_flag == last_ic_block && ic_tail) { in compute_ker()
291 kd_loop(ur_w, pad_l, pad_r, last_ic_block, handle_h_pad); in icb_loop()
4818 mov(reg_tmp, ptr[param + GET_OFF(last_ic_block)]); in generate()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_brgemm_1x1_conv.cpp | 163 const auto last_ic_block in init() local
168 ? (dim_t)rnd_up(jcp.ic, last_ic_block) * jcp.oc in init()
169 : (dim_t)rnd_up(jcp.ic, last_ic_block) * jcp.oc_block; in init()
170 wei_ocb_sz = jcp.wei_plain ? jcp.oc_block * last_ic_block in init()
|
| H A D | jit_brgemm_transpose_utils.hpp | 188 size_t last_ic_block; member
|
| H A D | jit_uni_x8s8s32x_deconvolution.hpp | 67 last_ic_block = 0x2U, enumerator
|
| H A D | jit_uni_x8s8s32x_conv_kernel.hpp | 57 last_ic_block, enumerator
|
| H A D | jit_avx512_core_x8s8s32x_conv_kernel.hpp | 59 last_ic_block, enumerator
|
| H A D | jit_avx512_core_x8s8s32x_deconvolution.hpp | 44 last_ic_block = 0x2U, enumerator
|
| H A D | jit_primitive_conf.hpp | 489 size_t last_ic_block; member
|
| H A D | jit_avx512_core_amx_convolution.cpp | 1254 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights_2d()
1492 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights_3d()
1740 p.last_ic_block = (nb_ic_blocks == jcp.nb_ic_blocking) ? 0 : 1; in compute_diff_weights()
1775 p.last_ic_block = ((ic_b + 1) >= jcp.nb_ic) ? 1 : 0; in reduce_and_convert_diff_weights_and_bias()
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 52 typedef enum { no_last_block, last_ic_block } ic_block_t; enumerator
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 84 const bool masked_write = ic_tail && last_ic_block_flag == last_ic_block; in compute_ker()
88 const int icb = (last_ic_block_flag == last_ic_block) in compute_ker()
120 if (jcp.is_relo && last_ic_block_flag == last_ic_block && ic_tail) { in compute_ker()
291 kd_loop(ur_w, pad_l, pad_r, last_ic_block, handle_h_pad); in icb_loop()
4844 mov(reg_tmp, ptr[param + GET_OFF(last_ic_block)]); in generate()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
| H A D | jit_sve_512_x8s8s32x_conv_kernel.hpp | 76 last_ic_block, enumerator
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
| H A D | jit_sve_512_x8s8s32x_conv_kernel.hpp | 76 last_ic_block, enumerator
|