Home
last modified time | relevance | path

Searched refs:tr_src_buf_size (Results 1 – 12 of 12) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_core_amx_convolution.cpp1123 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 Djit_avx512_core_bf16_convolution.cpp1061 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 Djit_primitive_conf.hpp152 size_t tr_src_buf_size, tr_src_buf_count; member
H A Djit_avx512_core_amx_conv_kernel.cpp3862 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 Djit_avx512_core_bf16_conv_kernel.cpp4480 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 Djit_avx512_core_amx_convolution.cpp1104 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 Djit_avx512_core_bf16_convolution.cpp1061 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 Djit_primitive_conf.hpp154 size_t tr_src_buf_size, tr_src_buf_count; member
H A Djit_avx512_core_amx_conv_kernel.cpp3888 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 Djit_avx512_core_bf16_conv_kernel.cpp4471 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 Djit_primitive_conf.hpp148 size_t tr_src_buf_size, tr_src_buf_count; member
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_primitive_conf.hpp148 size_t tr_src_buf_size, tr_src_buf_count; member