Searched refs:ow_nchunk (Results 1 – 3 of 3) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
| H A D | xe_lp_x8s8x_convolution.cpp | 113 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf() local
122 conf.gws_d[1] = conf.od * conf.oh * ow_nchunk; in init_conf()
144 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf() local
145 int ow_group = utils::max_div(ow_nchunk, max_ow_group); in init_conf()
222 int ow_nchunk = 1; in init_conf() local
238 ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf()
246 ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf()
383 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_kernel_ctx() local
384 int min_w_nchunk = nstl::min(ow_nchunk, iw_nchunk); in init_kernel_ctx()
449 kernel_ctx.define_int("OW_NCHUNK", ow_nchunk); in init_kernel_ctx()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
| H A D | xe_lp_x8s8x_convolution.cpp | 113 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf() local
122 conf.gws_d[1] = conf.od * conf.oh * ow_nchunk; in init_conf()
144 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf() local
145 int ow_group = utils::max_div(ow_nchunk, max_ow_group); in init_conf()
225 int ow_nchunk = 1; in init_conf() local
241 ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf()
249 ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf()
387 int ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_kernel_ctx() local
388 int min_w_nchunk = nstl::min(ow_nchunk, iw_nchunk); in init_kernel_ctx()
453 kernel_ctx.define_int("OW_NCHUNK", ow_nchunk); in init_kernel_ctx()
[all …]
|
| H A D | xe_hp_convolution.cpp | 68 int ow_nchunk; in init_conf() local
83 ow_nchunk = utils::div_up(conf.ow, conf.ow_block); in init_conf()
84 ow_group = nstl::min(utils::max_div(ow_nchunk, max_ow_group), 8); in init_conf()
|