Searched refs:is_ic_physically_blocked (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx2_convolution.cpp | 69 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_forward() local
71 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_forward()
72 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_forward()
228 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_backward_data() local
230 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_backward_data()
231 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_backward_data()
382 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_backward_weights() local
384 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_backward_weights()
385 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_backward_weights()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_avx2_convolution.cpp | 69 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_forward() local
71 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_forward()
72 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_forward()
228 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_backward_data() local
230 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_backward_data()
231 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_backward_data()
382 bool is_ic_physically_blocked = one_of(jcp.src_tag, format_tag::nCw8c, in execute_backward_weights() local
384 int g_ic_offset = is_ic_physically_blocked ? jcp.nb_ic : jcp.ic; in execute_backward_weights()
385 int icb_ic_scale = is_ic_physically_blocked ? 1 : jcp.ic_block; in execute_backward_weights()
|