Home
last modified time | relevance | path

Searched refs:loop_ngcdhw (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_1x1_conv.cpp355 else if (jcp.loop_order == loop_ngcdhw) in execute_forward_all()
391 else if (jcp.loop_order == loop_ngcdhw) in execute_forward_all()
H A Djit_brgemm_conv_utils.cpp631 else if (loop_order == loop_ngcdhw) in est_eff()
646 else if (loop_order == loop_ngcdhw) in est_eff()
1017 } else if (loop_order == loop_ngcdhw) { in est_eff_1x1()
1051 } else if (loop_order == loop_ngcdhw) { in est_eff_1x1()
1435 jcp.loop_order = (bcast_amount < wei_amount) ? loop_ngcdhw : loop_ndhwgc; in init_conf()
1615 jcp.loop_order = (bcast_amount < wei_amount) ? loop_ngcdhw : loop_ndhwgc; in init_1x1_conf()
H A Djit_primitive_conf.hpp779 loop_ngcdhw, enumerator
H A Djit_brgemm_conv.cpp585 else if (jcp.loop_order == loop_ngcdhw) in execute()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_brgemm_conv_utils.cpp786 else if (loop_order == loop_ngcdhw) in est_eff()
801 else if (loop_order == loop_ngcdhw) in est_eff()
1262 } else if (loop_order == loop_ngcdhw) { in est_eff_1x1()
1296 } else if (loop_order == loop_ngcdhw) { in est_eff_1x1()
1749 jcp.loop_order = (bcast_amount < wei_amount) ? loop_ngcdhw : loop_ndhwgc; in init_conf()
1965 jcp.loop_order = (bcast_amount < wei_amount) ? loop_ngcdhw : loop_ndhwgc; in init_1x1_conf()
1980 jcp.loop_order = max_size < wei_size ? loop_ngcdhw : loop_ndhwgc; in init_1x1_conf()
H A Djit_brgemm_1x1_conv.cpp505 else if (jcp.loop_order == loop_ngcdhw) in execute_forward_all()
543 else if (jcp.loop_order == loop_ngcdhw) in execute_forward_all()
H A Djit_primitive_conf.hpp820 loop_ngcdhw, enumerator
H A Djit_brgemm_conv.cpp648 else if (jcp.loop_order == loop_ngcdhw) in execute()
714 else if (jcp.loop_order == loop_ngcdhw) in execute()