Home
last modified time | relevance | path

Searched refs:ic_block_step_size (Results 1 – 4 of 4) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_avx2_conv_kernel_f32.hpp125 dim_t ic_block_step_size = jcp.kd * jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
126 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()
237 dim_t ic_block_step_size = jcp.kd * jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
238 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()
240 + i_ic_block * ic_block_step_size + ki * block_step_size in get_kernel_offset()
H A Djit_sse41_conv_kernel_f32.hpp109 dim_t ic_block_step_size = jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
110 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx2_conv_kernel_f32.hpp128 dim_t ic_block_step_size = jcp.kd * jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
129 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()
240 dim_t ic_block_step_size = jcp.kd * jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
241 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()
243 + i_ic_block * ic_block_step_size + ki * block_step_size in get_kernel_offset()
H A Djit_sse41_conv_kernel_f32.hpp110 dim_t ic_block_step_size = jcp.kh * jcp.kw * block_step_size; in get_kernel_offset() local
111 dim_t oc_block_step_size = jcp.nb_ic * ic_block_step_size; in get_kernel_offset()