/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx512_core_amx_convolution.cpp | 1123 return tr_src_buf_number(ti, g, ic) * adj * jcp.tr_src_buf_size in compute_diff_weights_2d() 1233 = tr_src + icb * jcp.tr_src_buf_size; in compute_diff_weights_2d() 1340 return tr_src_buf_number(ti, g, ic) * adj * jcp.tr_src_buf_size in compute_diff_weights_3d() 1469 = tr_src + icb * jcp.tr_src_buf_size; in compute_diff_weights_3d() 1552 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size * adj in compute_diff_weights() 1553 + ij * tr_row_size + nb_ic_block * jcp.tr_src_buf_size; in compute_diff_weights() 1560 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size * adj in compute_diff_weights() 1562 + nb_ic_block * jcp.tr_src_buf_size; in compute_diff_weights() 1581 + nb_oc_block * jcp.tr_src_buf_size; in compute_diff_weights() 1943 = &tr_src[ithr * jcp.tr_src_buf_size * jcp.nb_ic_blocking]; in prepare_scratchpad_data()
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1061 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights_2d() 1294 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights_3d() 1522 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights() 1530 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights() 1915 src_data_t *ts = &tr_src[ithr * jcp.tr_src_buf_size]; in prepare_scratchpad_data()
|
H A D | jit_primitive_conf.hpp | 152 size_t tr_src_buf_size, tr_src_buf_count; member
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 3862 dim_t icb_offset = jcp.typesize_in * icb * jcp.tr_src_buf_size; in compute_full_spat_loop() 4121 dim_t icb_offset = icb * jcp.tr_src_buf_size; in compute_ic_loop() 5108 jcp.tr_src_buf_size = jcp.tr_iw * jcp.ic_block * jcp.ih * jcp.id; in init_conf() 5133 = (jcp.tr_src_buf_count * jcp.tr_src_buf_size * jcp.nb_ic_blocking) in init_scratchpad()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 4480 jcp.tr_src_buf_size = jcp.tr_iw * jcp.ic_block * jcp.ih * jcp.id; in init_conf() 4507 const size_t tr_src_size = jcp.tr_src_buf_count * jcp.tr_src_buf_size in init_scratchpad()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_avx512_core_amx_convolution.cpp | 1104 return tr_src_buf_number(ti, g, ic) * adj * jcp.tr_src_buf_size in compute_diff_weights_2d() 1214 = tr_src + icb * jcp.tr_src_buf_size; in compute_diff_weights_2d() 1321 return tr_src_buf_number(ti, g, ic) * adj * jcp.tr_src_buf_size in compute_diff_weights_3d() 1450 = tr_src + icb * jcp.tr_src_buf_size; in compute_diff_weights_3d() 1533 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size * adj in compute_diff_weights() 1534 + ij * tr_row_size + nb_ic_block * jcp.tr_src_buf_size; in compute_diff_weights() 1541 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size * adj in compute_diff_weights() 1543 + nb_ic_block * jcp.tr_src_buf_size; in compute_diff_weights() 1562 + nb_oc_block * jcp.tr_src_buf_size; in compute_diff_weights() 1924 = &tr_src[ithr * jcp.tr_src_buf_size * jcp.nb_ic_blocking]; in prepare_scratchpad_data()
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1061 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights_2d() 1294 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights_3d() 1522 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights() 1530 return tr_src_buf_number(ti, g, ic) * jcp.tr_src_buf_size in compute_diff_weights() 1915 src_data_t *ts = &tr_src[ithr * jcp.tr_src_buf_size]; in prepare_scratchpad_data()
|
H A D | jit_primitive_conf.hpp | 154 size_t tr_src_buf_size, tr_src_buf_count; member
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 3888 dim_t icb_offset = jcp.typesize_in * icb * jcp.tr_src_buf_size; in compute_full_spat_loop() 4147 dim_t icb_offset = icb * jcp.tr_src_buf_size; in compute_ic_loop() 5134 jcp.tr_src_buf_size = jcp.tr_iw * jcp.ic_block * jcp.ih * jcp.id; in init_conf() 5159 = (jcp.tr_src_buf_count * jcp.tr_src_buf_size * jcp.nb_ic_blocking) in init_scratchpad()
|
H A D | jit_avx512_core_bf16_conv_kernel.cpp | 4471 jcp.tr_src_buf_size = jcp.tr_iw * jcp.ic_block * jcp.ih * jcp.id; in init_conf() 4498 const size_t tr_src_size = jcp.tr_src_buf_count * jcp.tr_src_buf_size in init_scratchpad()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 148 size_t tr_src_buf_size, tr_src_buf_count; member
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | jit_primitive_conf.hpp | 148 size_t tr_src_buf_size, tr_src_buf_count; member
|