/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | rnn.cpp | 179 && expect_dt(r.diff_dst_iter_c_desc, f32); in check_data_type_consistency_bwd() 240 CHECK(expect_dims(r.diff_dst_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency() 348 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags, in rnn_common_bwd_desc_init() 387 && xnor_md(dst_iter_c_desc, diff_dst_iter_c_desc); in rnn_common_bwd_desc_init() 397 diff_dst_iter_desc, diff_dst_iter_c_desc})); in rnn_common_bwd_desc_init() 428 maybe_init_md(rd.diff_dst_iter_c_desc, diff_dst_iter_c_desc); in rnn_common_bwd_desc_init() 604 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init() argument 636 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init_v2() argument 644 diff_dst_iter_c_desc, flags); in dnnl_lstm_backward_desc_init_v2() 670 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init_v3() argument [all …]
|
H A D | rnn_pd.hpp | 434 , diff_dst_iter_c_md_(desc_.diff_dst_iter_c_desc) {} in rnn_bwd_pd_t()
|
H A D | type_helpers.hpp | 556 && COMPARE_DESC_MEMBERS(diff_dst_iter_c_desc) in operator ==()
|
H A D | primitive_hashing.cpp | 592 seed = hash_combine(seed, get_md_hash(desc.diff_dst_iter_c_desc)); in get_desc_hash()
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | rnn.cpp | 180 && expect_dt(r.diff_dst_iter_c_desc, f32); in check_data_type_consistency_bwd() 241 CHECK(expect_dims(r.diff_dst_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency() 349 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags, in rnn_common_bwd_desc_init() 388 && xnor_md(dst_iter_c_desc, diff_dst_iter_c_desc); in rnn_common_bwd_desc_init() 398 diff_dst_iter_desc, diff_dst_iter_c_desc})); in rnn_common_bwd_desc_init() 429 maybe_init_md(rd.diff_dst_iter_c_desc, diff_dst_iter_c_desc); in rnn_common_bwd_desc_init() 605 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init() argument 637 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init_v2() argument 645 diff_dst_iter_c_desc, flags); in dnnl_lstm_backward_desc_init_v2() 671 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags) { in dnnl_lstm_backward_desc_init_v3() argument [all …]
|
H A D | rnn_pd.hpp | 434 , diff_dst_iter_c_md_(desc_.diff_dst_iter_c_desc) {} in rnn_bwd_pd_t()
|
H A D | type_helpers.hpp | 535 && COMPARE_DESC_MEMBERS(diff_dst_iter_c_desc) in operator ==()
|
H A D | primitive_hashing.cpp | 599 seed = hash_combine(seed, get_md_hash(desc.diff_dst_iter_c_desc)); in get_desc_hash()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/ |
H A D | dnnl.h | 2797 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 2885 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 2983 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags);
|
H A D | dnnl.hpp | 7671 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 8526 const memory::desc &diff_dst_iter_c_desc, in desc() 8546 &diff_dst_iter_c_desc.data, in desc() 8636 const memory::desc &diff_dst_iter_c_desc, in desc() 8654 &diff_dst_iter_c_desc.data, in desc() 8733 const memory::desc &diff_dst_iter_c_desc, in desc() 8748 &diff_dst_iter_c_desc.data, in desc() 8913 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 8914 return rnn_base::diff_dst_iter_c_desc(); in diff_dst_iter_c_desc()
|
H A D | dnnl_types.h | 2124 dnnl_memory_desc_t diff_dst_iter_c_desc; member
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/ |
H A D | dnnl.h | 2797 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 2885 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 2983 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags);
|
H A D | dnnl.hpp | 7671 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 8526 const memory::desc &diff_dst_iter_c_desc, in desc() 8546 &diff_dst_iter_c_desc.data, in desc() 8636 const memory::desc &diff_dst_iter_c_desc, in desc() 8654 &diff_dst_iter_c_desc.data, in desc() 8733 const memory::desc &diff_dst_iter_c_desc, in desc() 8748 &diff_dst_iter_c_desc.data, in desc() 8913 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 8914 return rnn_base::diff_dst_iter_c_desc(); in diff_dst_iter_c_desc()
|
H A D | dnnl_types.h | 2124 dnnl_memory_desc_t diff_dst_iter_c_desc; member
|
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/ |
H A D | dnnl.h | 2909 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 2997 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags); 3095 const dnnl_memory_desc_t *diff_dst_iter_c_desc, unsigned flags);
|
H A D | dnnl.hpp | 8123 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 8978 const memory::desc &diff_dst_iter_c_desc, in desc() 8998 &diff_dst_iter_c_desc.data, in desc() 9088 const memory::desc &diff_dst_iter_c_desc, in desc() 9106 &diff_dst_iter_c_desc.data, in desc() 9185 const memory::desc &diff_dst_iter_c_desc, in desc() 9200 &diff_dst_iter_c_desc.data, in desc() 9365 memory::desc diff_dst_iter_c_desc() const { in diff_dst_iter_c_desc() function 9366 return rnn_base::diff_dst_iter_c_desc(); in diff_dst_iter_c_desc()
|
H A D | dnnl_types.h | 2335 dnnl_memory_desc_t diff_dst_iter_c_desc; member
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/mkldnn_compat/examples/ |
H A D | rnn_training_f32.cpp | 500 leftmost_bwd_prim_desc.diff_dst_iter_c_desc(), eng); in simple_net()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/examples/ |
H A D | rnn_training_f32.cpp | 501 = dnnl::memory(leftmost_bwd_prim_desc.diff_dst_iter_c_desc(), eng); in simple_net()
|
/dports/math/onednn/oneDNN-2.5.1/examples/ |
H A D | rnn_training_f32.cpp | 501 = dnnl::memory(leftmost_bwd_prim_desc.diff_dst_iter_c_desc(), eng); in simple_net()
|