Home
last modified time | relevance | path

Searched refs:dnnl_backward_weights (Results 1 – 19 of 19) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/doc/
H A Dknobs_dir.md12 | BWD_WB | dnnl_backward_weights w/ bias
13 | BWD_W | dnnl_backward_weights w/o bias
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/doc/
H A Dknobs_dir.md12 | BWD_WB | dnnl_backward_weights w/ bias
13 | BWD_W | dnnl_backward_weights w/o bias
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_brgemm_inner_product_utils.cpp955 if (jbgp.prop_kind == dnnl_backward_weights in init_scratchpad()
967 } else if (jbgp.prop_kind == dnnl_backward_weights in init_scratchpad()
985 if (jbgp.use_buffer_a && jbgp.prop_kind == dnnl_backward_weights) { in init_scratchpad()
1004 if (jbgp.use_buffer_b && jbgp.prop_kind == dnnl_backward_weights) { in init_scratchpad()
1026 if (jbgp.prop_kind == dnnl_backward_weights && jbgp.with_bias in init_scratchpad()
1033 if (dnnl_thr_syncable() && jbgp.prop_kind == dnnl_backward_weights) in init_scratchpad()
H A Djit_brgemm_transpose_utils.cpp2078 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_src()
2081 else if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_src()
2107 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_to_vnni()
2111 else if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_to_vnni()
2140 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_amx_ip_trans_wei()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_brgemm_inner_product_utils.cpp925 if (jbgp.prop_kind == dnnl_backward_weights in init_scratchpad()
945 if (jbgp.use_buffer_a && jbgp.prop_kind == dnnl_backward_weights) { in init_scratchpad()
964 if (jbgp.use_buffer_b && jbgp.prop_kind == dnnl_backward_weights) { in init_scratchpad()
986 if (jbgp.prop_kind == dnnl_backward_weights && jbgp.with_bias in init_scratchpad()
993 if (dnnl_thr_syncable() && jbgp.prop_kind == dnnl_backward_weights) in init_scratchpad()
H A Djit_brgemm_transpose_utils.cpp1998 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_src()
2001 else if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_src()
2027 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_to_vnni()
2031 else if (conf->prop_kind == dnnl_backward_weights in create_brgemm_trans_to_vnni()
2060 if (conf->prop_kind == dnnl_backward_weights in create_brgemm_amx_ip_trans_wei()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/
H A Dmkldnn_dnnl_mangling.h311 #define mkldnn_backward_weights dnnl_backward_weights
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/
H A Dmkldnn_dnnl_mangling.h311 #define mkldnn_backward_weights dnnl_backward_weights
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Ddnnl_debug_autogenerated.cpp839 if (v == dnnl_backward_weights) return "backward_weights"; in dnnl_prop_kind2str()
H A Dc_types_map.hpp60 const prop_kind_t backward_weights = dnnl_backward_weights;
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/
H A Ddnnl_types.h1124 dnnl_backward_weights = 192, enumerator
H A Ddnnl.hpp456 backward_weights = dnnl_backward_weights,
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/
H A Ddnnl_types.h1124 dnnl_backward_weights = 192, enumerator
H A Ddnnl.hpp456 backward_weights = dnnl_backward_weights,
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Ddnnl_debug_autogenerated.cpp1050 if (v == dnnl_backward_weights) return "backward_weights"; in dnnl_prop_kind2str()
H A Dc_types_map.hpp60 const prop_kind_t backward_weights = dnnl_backward_weights;
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/
H A Ddnnl_types.h1335 dnnl_backward_weights = 192, enumerator
H A Ddnnl.hpp476 backward_weights = dnnl_backward_weights,
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/
H A Drnn_brgemm_utils.cpp1148 trans_conf.prop_kind = dnnl_backward_weights; in init_kernels()