| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_brgemm_conv.cpp | 669 btc.ohb = ohb; in execute()
679 auto oh_begin = ohb * jcp.oh_block; in execute()
707 last_ohb = ohb; in execute()
845 #define bmask(icb, odb, ohb, owb) \ in maybe_conv_inp() argument
852 && last_ohb == ohb && last_owb == owb) in maybe_conv_inp()
855 if (bmask(icb, odb, ohb, owb)) return; in maybe_conv_inp()
865 const auto prev_ohb = (jcp.copy_block_only || ohb == 0 in maybe_conv_inp()
872 || (odb > 0 && ohb > 0 in maybe_conv_inp()
879 const auto oh = ohb * jcp.oh_block; in maybe_conv_inp()
973 if (!jcp.copy_block_only) bmask(icb, odb, ohb, owb) = 1; in maybe_conv_inp()
[all …]
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 1159 int ohb, int ocb, size_t typesize) const { in get_out_ocb_offset() argument
1182 int ohb, int ocb) const { in get_wsp_ocb_offset() argument
1189 int ohb, int ocb, int j) const { in get_wsp_row_offset() argument
1190 return get_wsp_ocb_offset(ohb, ocb) in get_wsp_row_offset()
1566 for (int ohb = 0; ohb < h_blks; ohb++) { in store_output() local
1657 const int oh_index = ohb; in interleave_store()
1741 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1750 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1812 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1823 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
[all …]
|
| H A D | jit_brgemm_conv.hpp | 144 int od, odb, oh, ohb, owb; member
181 int g, int n, int icc, int odb, int ohb, int owb, int last_g,
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 349 size_t get_out_ocb_offset(int ohb, int ocb, size_t typesize) const;
350 size_t get_out_row_offset(int ohb, int ocb, int j, size_t typesize) const;
352 size_t get_wsp_ocb_offset(int ohb, int ocb) const;
353 size_t get_wsp_row_offset(int ohb, int ocb, int j) const;
357 size_t get_inp_offset(int ohb, int kw) const;
|
| H A D | gemm_bf16_convolution.cpp | 428 dim_t g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr_nspc() local
440 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
455 int oh = ohb * jcp.oh_block; in execute_forward_thr_nspc()
517 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
|
| H A D | jit_primitive_conf.hpp | 462 size_t ohb; member
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_brgemm_conv.cpp | 547 int n {0}, g {0}, ocb {0}, odb {0}, ohb {0}, owb {0}; \ in execute()
558 auto oh_begin = ohb * jcp.oh_blk_size; \ in execute()
568 g, n, icc, odb, ohb, owb); \ in execute()
583 BRGC_WO(n, jcp.mb, odb, jcp.nb_od, ohb, jcp.nb_oh, owb, jcp.nb_ow, g, in execute()
686 int g, int n, int icc, int odb, int ohb, int owb) const { in maybe_conv_inp() argument
691 #define bmask(icb, odb, ohb, owb) \ in maybe_conv_inp() argument
692 inp_buffer_mask[(((icb)*jcp.nb_od + (odb)) * jcp.nb_oh + (ohb)) \ in maybe_conv_inp()
696 if (bmask(icb, odb, ohb, owb)) return; in maybe_conv_inp()
703 = (ohb == 0 || bmask(icb, odb, ohb - 1, owb) == 0) ? false : true; in maybe_conv_inp()
706 = (odb > 0 && ohb > 0 && bmask(icb, odb - 1, ohb - 1, owb) == 0) in maybe_conv_inp()
[all …]
|
| H A D | jit_avx512_core_amx_conv_kernel.cpp | 1159 int ohb, int ocb, size_t typesize) const { in get_out_ocb_offset() argument
1182 int ohb, int ocb) const { in get_wsp_ocb_offset() argument
1189 int ohb, int ocb, int j) const { in get_wsp_row_offset() argument
1190 return get_wsp_ocb_offset(ohb, ocb) in get_wsp_row_offset()
1557 for (int ohb = 0; ohb < h_blks; ohb++) { in store_output() local
1648 const int oh_index = ohb; in interleave_store()
1732 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1741 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1803 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
1814 for (int ohb = 0; ohb < jcp.nb_oh_blocking; ohb++) { in compute_icb_loop() local
[all …]
|
| H A D | jit_avx512_core_amx_conv_kernel.hpp | 347 size_t get_out_ocb_offset(int ohb, int ocb, size_t typesize) const;
348 size_t get_out_row_offset(int ohb, int ocb, int j, size_t typesize) const;
350 size_t get_wsp_ocb_offset(int ohb, int ocb) const;
351 size_t get_wsp_row_offset(int ohb, int ocb, int j) const;
355 size_t get_inp_offset(int ohb, int kw) const;
|
| H A D | jit_brgemm_conv.hpp | 158 int ohb, int owb) const;
|
| H A D | gemm_bf16_convolution.cpp | 438 int g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr_nspc() local
450 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
465 int oh = ohb * jcp.oh_block; in execute_forward_thr_nspc()
527 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
|
| H A D | jit_primitive_conf.hpp | 460 size_t ohb; member
|
| H A D | jit_avx512_core_amx_convolution.cpp | 380 p.ohb = limit_idx; in execute_forward_reduced_lowering()
764 p.ohb = limit_idx; in execute_forward()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | gemm_x8s8s32x_convolution.cpp | 200 int g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr() local
212 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr()
219 const int oh = ohb * jcp.oh_block; in execute_forward_thr()
289 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr()
|
| H A D | gemm_convolution.cpp | 91 int g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr_nspc() local
106 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
116 int oh = ohb * jcp.oh_block; in execute_forward_thr_nspc()
218 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | gemm_x8s8s32x_convolution.cpp | 196 dim_t g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr() local
208 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr()
215 const int oh = ohb * jcp.oh_block; in execute_forward_thr()
300 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr()
|
| H A D | gemm_convolution.cpp | 89 dim_t g {0}, n {0}, ohb {0}, owb {0}; in execute_forward_thr_nspc() local
103 nd_iterator_init(start, n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
113 int oh = ohb * jcp.oh_block; in execute_forward_thr_nspc()
215 nd_iterator_step(n, jcp.mb, g, jcp.ngroups, ohb, nb_oh, owb, nb_ow); in execute_forward_thr_nspc()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
| H A D | primitive_conf.hpp | 279 int odb, ohb, owb; member
1362 conf.ohb = 1; in bwd_w_compute_block_sizes()
1367 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
1392 conf.ohb = 1; in bwd_w_compute_block_sizes()
1413 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
1421 int ihb = conf.ndims < 4 ? 1 : conf.ohb + 2 * (conf.kh - 1); in bwd_w_compute_block_sizes()
1425 size_t ospb = conf.owb * conf.ohb * conf.odb; in bwd_w_compute_block_sizes()
1469 int ohb_next = next_candidate(conf.oh, conf.ohb); in bwd_w_compute_block_sizes()
1470 try_next(conf.ohb, ohb_next); in bwd_w_compute_block_sizes()
1483 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | gen9_convolution.cpp | 710 conf.ohb = 1; in bwd_w_compute_block_sizes()
715 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
740 conf.ohb = 1; in bwd_w_compute_block_sizes()
761 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
769 int ihb = conf.ndims < 4 ? 1 : conf.ohb + 2 * (conf.kh - 1); in bwd_w_compute_block_sizes()
773 size_t ospb = conf.owb * conf.ohb * conf.odb; in bwd_w_compute_block_sizes()
817 int ohb_next = next_candidate(conf.oh, conf.ohb); in bwd_w_compute_block_sizes()
818 try_next(conf.ohb, ohb_next); in bwd_w_compute_block_sizes()
831 * utils::div_up(conf.oh, conf.ohb) in bwd_w_compute_block_sizes()
1087 kernel_ctx.define_int("OHB", conf.ohb); in init_kernel_ctx()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | xe_hp_1st_bwd_convolution.cpp | 223 kernel_ctx.define_int("OHB", conf.ohb); in init_kernel_ctx()
|
| /dports/science/gromacs/gromacs-2021.4/src/gromacs/gmxana/ |
| H A D | gmx_hbond.cpp | 1564 int i, j, j0, k, m, nh, ihb, ohb, nhydro, ndump = 0; in do_hblife() local
1606 ohb = 0; in do_hblife()
1616 if (ihb != ohb) in do_hblife()
1626 ohb = ihb; in do_hblife()
|
| /dports/audio/espeak/espeak-1.48.04-source/dictsource/ |
| H A D | ms_list | 51 _17 t'udZ%ohb@las
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
| H A D | primitive_conf.hpp | 271 int odb, ohb, owb; member
|
| /dports/japanese/canna-lib/Canna37p3/dic/phono/ |
| H A D | tut.kpdef | 1950 ohb ��
|
| /dports/japanese/canna-server/Canna37p3/dic/phono/ |
| H A D | tut.kpdef | 1950 ohb ��
|