Home
last modified time | relevance | path

Searched refs:diff_dst_iter_c_desc (Results 1 – 20 of 20) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn.cpp179 && 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 Drnn_pd.hpp434 , diff_dst_iter_c_md_(desc_.diff_dst_iter_c_desc) {} in rnn_bwd_pd_t()
H A Dtype_helpers.hpp556 && COMPARE_DESC_MEMBERS(diff_dst_iter_c_desc) in operator ==()
H A Dprimitive_hashing.cpp592 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 Drnn.cpp180 && 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 Drnn_pd.hpp434 , diff_dst_iter_c_md_(desc_.diff_dst_iter_c_desc) {} in rnn_bwd_pd_t()
H A Dtype_helpers.hpp535 && COMPARE_DESC_MEMBERS(diff_dst_iter_c_desc) in operator ==()
H A Dprimitive_hashing.cpp599 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 Ddnnl.h2797 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 Ddnnl.hpp7671 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 Ddnnl_types.h2124 dnnl_memory_desc_t diff_dst_iter_c_desc; member
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/
H A Ddnnl.h2797 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 Ddnnl.hpp7671 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 Ddnnl_types.h2124 dnnl_memory_desc_t diff_dst_iter_c_desc; member
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/
H A Ddnnl.h2909 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 Ddnnl.hpp8123 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 Ddnnl_types.h2335 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 Drnn_training_f32.cpp500 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 Drnn_training_f32.cpp501 = 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 Drnn_training_f32.cpp501 = dnnl::memory(leftmost_bwd_prim_desc.diff_dst_iter_c_desc(), eng); in simple_net()