/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | rnn_pd.hpp | 316 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 D | rnn_pd.hpp | 316 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 D | verbose.cpp | 867 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 D | rnn_training_f32.cpp | 684 {DNNL_ARG_DIFF_DST_ITER, leftmost_diff_dst_iter_memory}, in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/examples/ |
H A D | rnn_training_f32.cpp | 684 {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 D | rnn.cpp | 1022 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 D | rnn.cpp | 1029 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 D | mkldnn_dnnl_mangling.h | 31 #define MKLDNN_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_ITER
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/ |
H A D | mkldnn_dnnl_mangling.h | 31 #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 D | dnnl_types.h | 2524 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
|
H A D | dnnl.hpp | 7666 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 D | dnnl_types.h | 2524 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
|
H A D | dnnl.hpp | 7666 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 D | rnn.md | 325 | \diffdstiter | DNNL_ARG_DIFF_DST_ITER |
|
/dports/math/onednn/oneDNN-2.5.1/doc/primitives/ |
H A D | rnn.md | 325 | \diffdstiter | DNNL_ARG_DIFF_DST_ITER |
|
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 2747 #define DNNL_ARG_DIFF_DST_ITER DNNL_ARG_DIFF_DST_1 macro
|
H A D | dnnl.hpp | 8118 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 D | ref_rnn.cpp | 1129 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 D | ref_rnn.cpp | 1188 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 D | ref_rnn.cpp | 1395 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 D | ref_rnn.cpp | 1427 auto &diff_dst_iter_native_ = CTX_IN_STORAGE(DNNL_ARG_DIFF_DST_ITER); in execute_()
|