Home
last modified time | relevance | path

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 Djit_brgemm_conv.hpp162 int i_N, int init_bcast_dim, int po_bcast_dim, bool need_postwork);
190 bool need_postwork; member
H A Djit_brgemm_conv.cpp263 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 Djit_brgemm_1x1_conv.hpp128 bool need_postwork; member
H A Djit_brgemm_1x1_conv.cpp171 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 Djit_brgemm_conv.hpp187 int i_N, int init_bcast_dim, int po_bcast_dim, bool need_postwork);
218 bool need_postwork; member
H A Djit_brgemm_conv.cpp322 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 Djit_brgemm_1x1_conv.hpp136 bool need_postwork; member
H A Djit_brgemm_1x1_conv.cpp173 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()