Home
last modified time | relevance | path

Searched refs:DNNL_ARG_DST_ITER (Results 1 – 25 of 31) sorted by relevance

12

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn_pd.hpp220 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 Drnn_pd.hpp220 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 Dlstm.cpp180 lstm_args.insert({DNNL_ARG_DST_ITER, dst_iter_mem}); in lstm_example()
/dports/math/onednn/oneDNN-2.5.1/examples/primitives/
H A Dlstm.cpp180 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 Drnn_training_f32.cpp390 {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 Dcpu_rnn_inference_f32.cpp684 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
H A Dcpu_rnn_inference_int8.cpp763 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
/dports/math/onednn/oneDNN-2.5.1/examples/
H A Drnn_training_f32.cpp390 {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 Dcpu_rnn_inference_f32.cpp684 {DNNL_ARG_DST_ITER, dec_dst_iter_memory}, in simple_net()
H A Dcpu_rnn_inference_int8.cpp763 {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 Dtest_rnn_forward.cpp128 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 Dtest_iface_runtime_attr.cpp418 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 Dtest_rnn_forward.cpp128 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 Dtest_iface_runtime_attr.cpp405 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 Drnn.cpp879 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 Drnn.cpp886 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 Dmkldnn_dnnl_mangling.h51 #define MKLDNN_ARG_DST_ITER DNNL_ARG_DST_ITER
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/
H A Dmkldnn_dnnl_mangling.h51 #define MKLDNN_ARG_DST_ITER DNNL_ARG_DST_ITER
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/
H A Dref_rnn.cpp1121 ? 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 Dref_rnn.cpp1182 ? 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 Dref_rnn.cpp1388 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 Dref_rnn.cpp1420 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 Ddnnl_types.h2433 #define DNNL_ARG_DST_ITER DNNL_ARG_DST_1 macro
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/
H A Ddnnl_types.h2433 #define DNNL_ARG_DST_ITER DNNL_ARG_DST_1 macro
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/primitives/
H A Drnn.md313 | \dstiter | DNNL_ARG_DST_ITER |

12