Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx512_common_conv_winograd_kernel_f32.cpp419 && (dimN_reg_block < jcp.nb_reg) in set_wsched_DATA_W_S_G_D_avx512_common()
425 if (jcp.dimN_reg_block >= jcp.nb_reg) { in set_wsched_DATA_W_S_G_D_avx512_common()
429 return (dimN_reg_block < jcp.nb_reg) in set_wsched_DATA_W_S_G_D_avx512_common()
515 jcp.nb_reg = 32 - jcp.zmm_start; in init_conf_kernel()
1129 jcp.nb_reg = 32 - jcp.zmm_start; in init_conf()
H A Djit_primitive_conf.hpp391 int nb_reg; member
H A Djit_avx512_core_f32_wino_conv_4x3_kernel.cpp1210 ? dimN_reg_block < jcp.nb_reg && dimN_reg_block > current_best in set_kernel_dims_reg_block()
1214 < jcp.nb_reg in set_kernel_dims_reg_block()
1368 jcp.nb_reg = 32; in init_conf_kernel()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx512_common_conv_winograd_kernel_f32.cpp419 && (dimN_reg_block < jcp.nb_reg) in set_wsched_DATA_W_S_G_D_avx512_common()
425 if (jcp.dimN_reg_block >= jcp.nb_reg) { in set_wsched_DATA_W_S_G_D_avx512_common()
429 return (dimN_reg_block < jcp.nb_reg) in set_wsched_DATA_W_S_G_D_avx512_common()
515 jcp.nb_reg = 32 - jcp.zmm_start; in init_conf_kernel()
1129 jcp.nb_reg = 32 - jcp.zmm_start; in init_conf()
H A Djit_primitive_conf.hpp393 int nb_reg; member
H A Djit_avx512_core_f32_wino_conv_4x3_kernel.cpp1210 ? dimN_reg_block < jcp.nb_reg && dimN_reg_block > current_best in set_kernel_dims_reg_block()
1214 < jcp.nb_reg in set_kernel_dims_reg_block()
1368 jcp.nb_reg = 32; in init_conf_kernel()