Home
last modified time | relevance | path

Searched refs:DNNL_ARG_DIFF_DST_ITER (Results 1 – 21 of 21) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn_pd.hpp316 arg, DNNL_ARG_DST_ITER, DNNL_ARG_DIFF_DST_ITER)) in arg_usage()
362 case DNNL_ARG_DIFF_DST_ITER: return diff_dst_md(1); in arg_md()
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Drnn_pd.hpp316 arg, DNNL_ARG_DST_ITER, DNNL_ARG_DIFF_DST_ITER)) in arg_usage()
362 case DNNL_ARG_DIFF_DST_ITER: return diff_dst_md(1); in arg_md()
H A Dverbose.cpp867 pd->with_dst_iter(), DNNL_ARG_DIFF_DST_ITER, "diff_dst_iter"); in init_info_rnn()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/
H A Drnn_training_f32.cpp684 {DNNL_ARG_DIFF_DST_ITER, leftmost_diff_dst_iter_memory}, in simple_net()
/dports/math/onednn/oneDNN-2.5.1/examples/
H A Drnn_training_f32.cpp684 {DNNL_ARG_DIFF_DST_ITER, leftmost_diff_dst_iter_memory}, in simple_net()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Drnn.cpp1022 const auto &diff_dst_iter_md = q(const_bpd, DNNL_ARG_DIFF_DST_ITER); in doit()
1120 args.set(DNNL_ARG_DIFF_DST_ITER, diff_dst_iter_dt); in doit()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/
H A Drnn.cpp1029 const auto &diff_dst_iter_md = q(const_bpd, DNNL_ARG_DIFF_DST_ITER); in doit()
1127 args.set(DNNL_ARG_DIFF_DST_ITER, diff_dst_iter_dt); in doit()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/
H A Dmkldnn_dnnl_mangling.h31 #define MKLDNN_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_ITER
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/
H A Dmkldnn_dnnl_mangling.h31 #define MKLDNN_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_ITER
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/
H A Ddnnl_types.h2524 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
H A Ddnnl.hpp7666 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_DST_ITER); in diff_dst_iter_desc()
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/
H A Ddnnl_types.h2524 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
H A Ddnnl.hpp7666 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_DST_ITER); in diff_dst_iter_desc()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/primitives/
H A Drnn.md325 | \diffdstiter | DNNL_ARG_DIFF_DST_ITER |
/dports/math/onednn/oneDNN-2.5.1/doc/primitives/
H A Drnn.md325 | \diffdstiter | DNNL_ARG_DIFF_DST_ITER |
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/
H A Ddnnl_types.h2747 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
H A Ddnnl.hpp8118 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_DST_ITER); in diff_dst_iter_desc()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/
H A Dref_rnn.cpp1129 auto diff_dst_iter = CTX_IN_MEM(const gemm_acc_t *, DNNL_ARG_DIFF_DST_ITER); in execute_()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/
H A Dref_rnn.cpp1188 auto diff_dst_iter = CTX_IN_MEM(const gemm_acc_t *, DNNL_ARG_DIFF_DST_ITER); in execute_()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/
H A Dref_rnn.cpp1395 auto &diff_dst_iter_native_ = CTX_IN_STORAGE(DNNL_ARG_DIFF_DST_ITER); in execute_()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/
H A Dref_rnn.cpp1427 auto &diff_dst_iter_native_ = CTX_IN_STORAGE(DNNL_ARG_DIFF_DST_ITER); in execute_()