Home
last modified time | relevance | path

Searched refs:ow_blocks (Results 1 – 6 of 6) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Dgen9_wino_convolution.cpp40 int ow_blocks = conf.wino_ow / conf.ow_block; in is_impl_optimal() local
48 int blocks = ow_blocks * oh_blocks * oc_blocks; in is_impl_optimal()
60 score = (float)conf.oc * conf.ic / (oh_blocks * ow_blocks); in is_impl_optimal()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Dgen9_wino_convolution.cpp40 int ow_blocks = conf.wino_ow / conf.ow_block; in is_impl_optimal() local
48 int blocks = ow_blocks * oh_blocks * oc_blocks; in is_impl_optimal()
60 score = (float)conf.oc * conf.ic / (oh_blocks * ow_blocks); in is_impl_optimal()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_primitive_conf.hpp247 int iw_blocks, ow_blocks; member
H A Djit_avx512_core_amx_conv_kernel.cpp1946 true, jcp.ow_blocks, jcp.l_pad_output, ow_r_pad_start); in compute_ow_loop()
1954 const int last_owb_tile_blocks = jcp.ow_blocks % ow_blocks_per_call == 0 in compute_ow_loop()
1956 : jcp.ow_blocks % ow_blocks_per_call; in compute_ow_loop()
1998 int last_owb_tile_blocks = jcp.ow_blocks % ow_blocks_per_call; in compute_ow_loop()
2492 jcp.ow_blocks = utils::div_up(jcp.ow, jcp.tile_width); in init_conf()
2496 if (jcp.oh_per_tile == 1 && jcp.ow % jcp.ow_blocks == 0) in init_conf()
2497 jcp.tile_width = jcp.ow / jcp.ow_blocks; in init_conf()
2505 IMPLICATION(jcp.ow_blocks > 1, jcp.oh % 2 == 0), in init_conf()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_primitive_conf.hpp249 int iw_blocks, ow_blocks; member
H A Djit_avx512_core_amx_conv_kernel.cpp1955 true, jcp.ow_blocks, jcp.l_pad_output, ow_r_pad_start); in compute_ow_loop()
1963 const int last_owb_tile_blocks = jcp.ow_blocks % ow_blocks_per_call == 0 in compute_ow_loop()
1965 : jcp.ow_blocks % ow_blocks_per_call; in compute_ow_loop()
2007 int last_owb_tile_blocks = jcp.ow_blocks % ow_blocks_per_call; in compute_ow_loop()
2505 jcp.ow_blocks = utils::div_up(jcp.ow, jcp.tile_width); in init_conf()
2509 if (jcp.oh_per_tile == 1 && jcp.ow % jcp.ow_blocks == 0) in init_conf()
2510 jcp.tile_width = jcp.ow / jcp.ow_blocks; in init_conf()
2518 IMPLICATION(jcp.ow_blocks > 1, jcp.oh % 2 == 0), in init_conf()