Searched refs:DNNL_ARG_DIFF_SRC_ITER_C (Results 1 – 20 of 20) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | rnn_pd.hpp | 311 if (arg == DNNL_ARG_DIFF_SRC_ITER_C) return arg_usage_t::output; in arg_usage() 336 case DNNL_ARG_DIFF_SRC_ITER_C: return diff_src_md(2); in arg_md()
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | rnn_pd.hpp | 311 if (arg == DNNL_ARG_DIFF_SRC_ITER_C) return arg_usage_t::output; in arg_usage() 336 case DNNL_ARG_DIFF_SRC_ITER_C: return diff_src_md(2); in arg_md()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
H A D | rnn_training_f32.cpp | 658 {DNNL_ARG_DIFF_SRC_ITER_C, in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/examples/ |
H A D | rnn_training_f32.cpp | 658 {DNNL_ARG_DIFF_SRC_ITER_C, in simple_net()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 1011 const auto &diff_src_iter_c_md = q(const_bpd, DNNL_ARG_DIFF_SRC_ITER_C); in doit() 1125 args.set(DNNL_ARG_DIFF_SRC_ITER_C, diff_src_iter_c_dt); in doit()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | rnn.cpp | 1018 const auto &diff_src_iter_c_md = q(const_bpd, DNNL_ARG_DIFF_SRC_ITER_C); in doit() 1132 args.set(DNNL_ARG_DIFF_SRC_ITER_C, diff_src_iter_c_dt); in doit()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/ |
H A D | mkldnn_dnnl_mangling.h | 40 #define MKLDNN_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_ITER_C
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/ |
H A D | mkldnn_dnnl_mangling.h | 40 #define MKLDNN_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_ITER_C
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 2509 #define DNNL_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_2 macro
|
H A D | dnnl.hpp | 7618 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_SRC_ITER_C); in diff_src_iter_c_desc()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/ |
H A D | dnnl_types.h | 2509 #define DNNL_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_2 macro
|
H A D | dnnl.hpp | 7618 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_SRC_ITER_C); in diff_src_iter_c_desc()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/primitives/ |
H A D | rnn.md | 318 | \diffsrciterc | DNNL_ARG_DIFF_SRC_ITER_C |
|
/dports/math/onednn/oneDNN-2.5.1/doc/primitives/ |
H A D | rnn.md | 318 | \diffsrciterc | DNNL_ARG_DIFF_SRC_ITER_C |
|
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/ |
H A D | dnnl_types.h | 2732 #define DNNL_ARG_DIFF_SRC_ITER_C DNNL_ARG_DIFF_SRC_2 macro
|
H A D | dnnl.hpp | 8070 return base::query_md(query::exec_arg_md, DNNL_ARG_DIFF_SRC_ITER_C); in diff_src_iter_c_desc()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 1206 auto diff_src_iter_c = CTX_OUT_MEM(float *, DNNL_ARG_DIFF_SRC_ITER_C); in execute_()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 1265 auto diff_src_iter_c = CTX_OUT_MEM(float *, DNNL_ARG_DIFF_SRC_ITER_C); in execute_()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
H A D | ref_rnn.cpp | 1420 auto &diff_src_iter_c_native_ = CTX_OUT_STORAGE(DNNL_ARG_DIFF_SRC_ITER_C); in execute_()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
H A D | ref_rnn.cpp | 1461 auto &diff_src_iter_c_native_ = CTX_OUT_STORAGE(DNNL_ARG_DIFF_SRC_ITER_C); in execute_()
|