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 D | gen9_wino_convolution.cpp | 40 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 D | gen9_wino_convolution.cpp | 40 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 D | jit_primitive_conf.hpp | 247 int iw_blocks, ow_blocks; member
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 1946 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 D | jit_primitive_conf.hpp | 249 int iw_blocks, ow_blocks; member
|
H A D | jit_avx512_core_amx_conv_kernel.cpp | 1955 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()
|