Searched refs:need_postwork (Results 1 – 8 of 8) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_brgemm_conv.hpp | 162 int i_N, int init_bcast_dim, int po_bcast_dim, bool need_postwork); 190 bool need_postwork; member
|
H A D | jit_brgemm_conv.cpp | 263 bool need_postwork) { in add_po_kernels() argument 279 if ((need_postwork || jcp.use_buffer) && po_bcast_dim > 0) { in add_po_kernels() 351 need_postwork = jcp.with_bias || jcp.with_eltwise || jcp.with_binary in init() 389 add_po_kernels(i_N, M, M, need_postwork); in init() 424 i_N, init_bcast_dim, po_bcast_dim, need_postwork); in init() 459 i_N, init_bcast_dim, po_bcast_dim, need_postwork); in init() 870 const auto do_postwork = need_postwork && icc == (ic_chunks - 1) in ker_base() 962 const auto do_postwork = need_postwork && icc == (ic_chunks - 1); in ker_base() 1048 const auto do_postwork = need_postwork && icc == (ic_chunks - 1) in ker_trans() 1077 const auto do_postwork = need_postwork && icc == (ic_chunks - 1); in ker_trans() [all …]
|
H A D | jit_brgemm_1x1_conv.hpp | 128 bool need_postwork; member
|
H A D | jit_brgemm_1x1_conv.cpp | 171 need_postwork = jcp.with_bias || jcp.with_eltwise || jcp.with_binary in init() 285 = (need_postwork || jcp.use_buffer) && icc == ic_chunks - 1; in exec_ker()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_brgemm_conv.hpp | 187 int i_N, int init_bcast_dim, int po_bcast_dim, bool need_postwork); 218 bool need_postwork; member
|
H A D | jit_brgemm_conv.cpp | 322 int i_N, int init_bcast_dim, int po_bcast_dim, bool need_postwork) { in add_po_kernels() argument 341 if ((need_postwork || jcp.use_buffer) && po_bcast_dim > 0) { in add_po_kernels() 420 need_postwork = jcp.with_bias || jcp.with_eltwise || jcp.with_binary in init() 478 add_po_kernels(i_N, M, M, need_postwork); in init() 514 i_N, init_bcast_dim, po_bcast_dim, need_postwork); in init() 550 i_N, init_bcast_dim, po_bcast_dim, need_postwork); in init() 1087 const auto do_postwork = need_postwork && btc.icc == (ic_chunks - 1) in ker_base() 1178 const auto do_postwork = need_postwork && btc.icc == (ic_chunks - 1); in ker_base() 1284 const auto do_postwork = need_postwork && btc.icc == (ic_chunks - 1) in ker_trans() 1324 const auto do_postwork = need_postwork && btc.icc == (ic_chunks - 1); in ker_trans() [all …]
|
H A D | jit_brgemm_1x1_conv.hpp | 136 bool need_postwork; member
|
H A D | jit_brgemm_1x1_conv.cpp | 173 need_postwork = jcp.with_bias || jcp.with_eltwise || jcp.with_binary in init() 406 = (need_postwork || jcp.use_buffer) && icc == ic_chunks - 1; in exec_ker()
|