/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | rnn_pd.hpp | 220 if (arg == DNNL_ARG_DST_ITER && with_dst_iter()) in arg_usage() 248 case DNNL_ARG_DST_ITER: return dst_md(1); in arg_md() 316 arg, DNNL_ARG_DST_ITER, DNNL_ARG_DIFF_DST_ITER)) in arg_usage() 359 case DNNL_ARG_DST_ITER: return dst_md(1); in arg_md()
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | rnn_pd.hpp | 220 if (arg == DNNL_ARG_DST_ITER && with_dst_iter()) in arg_usage() 248 case DNNL_ARG_DST_ITER: return dst_md(1); in arg_md() 316 arg, DNNL_ARG_DST_ITER, DNNL_ARG_DIFF_DST_ITER)) in arg_usage() 359 case DNNL_ARG_DST_ITER: return dst_md(1); in arg_md()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/primitives/ |
H A D | lstm.cpp | 180 lstm_args.insert({DNNL_ARG_DST_ITER, dst_iter_mem}); in lstm_example()
|
/dports/math/onednn/oneDNN-2.5.1/examples/primitives/ |
H A D | lstm.cpp | 180 lstm_args.insert({DNNL_ARG_DST_ITER, dst_iter_mem}); in lstm_example()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
H A D | rnn_training_f32.cpp | 390 {DNNL_ARG_DST_ITER, leftmost_dst_iter_memory}, in simple_net() 675 {DNNL_ARG_DST_ITER, leftmost_dst_iter_bwd_memory}, in simple_net()
|
H A D | cpu_rnn_inference_f32.cpp | 684 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
|
H A D | cpu_rnn_inference_int8.cpp | 763 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/examples/ |
H A D | rnn_training_f32.cpp | 390 {DNNL_ARG_DST_ITER, leftmost_dst_iter_memory}, in simple_net() 675 {DNNL_ARG_DST_ITER, leftmost_dst_iter_bwd_memory}, in simple_net()
|
H A D | cpu_rnn_inference_f32.cpp | 684 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
|
H A D | cpu_rnn_inference_int8.cpp | 763 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_rnn_forward.cpp | 128 ASSERT_TRUE(pd.query_md(query::exec_arg_md, DNNL_ARG_DST_ITER) in testExecArgQueries() 373 {DNNL_ARG_DST_ITER, dst_iter_ref}, in Test() 397 {DNNL_ARG_DST_ITER, dst_iter_tgt}, in Test()
|
H A D | test_iface_runtime_attr.cpp | 418 DNNL_ARG_DST_LAYER, DNNL_ARG_DST_ITER, DNNL_ARG_DST_ITER_C}) { in TEST_F()
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_rnn_forward.cpp | 128 ASSERT_TRUE(pd.query_md(query::exec_arg_md, DNNL_ARG_DST_ITER) in testExecArgQueries() 373 {DNNL_ARG_DST_ITER, dst_iter_ref}, in Test() 397 {DNNL_ARG_DST_ITER, dst_iter_tgt}, in Test()
|
H A D | test_iface_runtime_attr.cpp | 405 DNNL_ARG_DST_LAYER, DNNL_ARG_DST_ITER, DNNL_ARG_DST_ITER_C}) { in TEST_F()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 879 const auto &dst_iter_md = q(const_fpd, DNNL_ARG_DST_ITER); in doit() 973 args.set(DNNL_ARG_DST_ITER, dst_iter_dt); in doit() 1117 args.set(DNNL_ARG_DST_ITER, dst_iter_dt); in doit()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 886 const auto &dst_iter_md = q(const_fpd, DNNL_ARG_DST_ITER); in doit() 980 args.set(DNNL_ARG_DST_ITER, dst_iter_dt); in doit() 1124 args.set(DNNL_ARG_DST_ITER, dst_iter_dt); in doit()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/ |
H A D | mkldnn_dnnl_mangling.h | 51 #define MKLDNN_ARG_DST_ITER DNNL_ARG_DST_ITER
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/ |
H A D | mkldnn_dnnl_mangling.h | 51 #define MKLDNN_ARG_DST_ITER DNNL_ARG_DST_ITER
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 1121 ? CTX_OUT_MEM(char *, DNNL_ARG_DST_ITER) in execute_() 1122 : const_cast<char *>(CTX_IN_MEM(const char *, DNNL_ARG_DST_ITER)); in execute_()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 1182 ? CTX_OUT_MEM(char *, DNNL_ARG_DST_ITER) in execute_() 1183 : const_cast<char *>(CTX_IN_MEM(const char *, DNNL_ARG_DST_ITER)); in execute_()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
H A D | ref_rnn.cpp | 1388 auto &dst_last_iter_native_ = is_fwd ? CTX_OUT_STORAGE(DNNL_ARG_DST_ITER) in execute_() 1389 : CTX_IN_STORAGE(DNNL_ARG_DST_ITER); in execute_()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
H A D | ref_rnn.cpp | 1420 auto &dst_last_iter_native_ = is_fwd ? CTX_OUT_STORAGE(DNNL_ARG_DST_ITER) in execute_() 1421 : CTX_IN_STORAGE(DNNL_ARG_DST_ITER); in execute_()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 2433 #define DNNL_ARG_DST_ITER DNNL_ARG_DST_1 macro
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/ |
H A D | dnnl_types.h | 2433 #define DNNL_ARG_DST_ITER DNNL_ARG_DST_1 macro
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/primitives/ |
H A D | rnn.md | 313 | \dstiter | DNNL_ARG_DST_ITER |
|