Home
last modified time | relevance | path

Searched refs:diff_weights_peephole_desc (Results 1 – 17 of 17) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn.cpp175 && expect_dt(r.diff_weights_peephole_desc, f32) in check_data_type_consistency_bwd()
235 CHECK(expect_dims(r.diff_weights_peephole_desc, {L, D, 3, DHC})); in check_dim_consistency()
343 const dnnl_memory_desc_t *diff_weights_peephole_desc, in rnn_common_bwd_desc_init() argument
382 && xnor_md(weights_peephole_desc, diff_weights_peephole_desc) in rnn_common_bwd_desc_init()
395 diff_weights_iter_desc, diff_weights_peephole_desc, in rnn_common_bwd_desc_init()
422 maybe_init_md(rd.diff_weights_peephole_desc, diff_weights_peephole_desc); in rnn_common_bwd_desc_init()
632 const dnnl_memory_desc_t *diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v2() argument
642 diff_weights_iter_desc, diff_weights_peephole_desc, nullptr, in dnnl_lstm_backward_desc_init_v2()
665 const dnnl_memory_desc_t *diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v3() argument
677 diff_weights_iter_desc, diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v3()
H A Drnn_pd.hpp429 , diff_weights_peephole_md_(desc_.diff_weights_peephole_desc) in rnn_bwd_pd_t()
H A Dtype_helpers.hpp557 && COMPARE_DESC_MEMBERS(diff_weights_peephole_desc) in operator ==()
H A Dprimitive_hashing.cpp593 seed = hash_combine(seed, get_md_hash(desc.diff_weights_peephole_desc)); in get_desc_hash()
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Drnn.cpp176 && expect_dt(r.diff_weights_peephole_desc, f32) in check_data_type_consistency_bwd()
236 CHECK(expect_dims(r.diff_weights_peephole_desc, {L, D, 3, DHC})); in check_dim_consistency()
344 const dnnl_memory_desc_t *diff_weights_peephole_desc, in rnn_common_bwd_desc_init() argument
383 && xnor_md(weights_peephole_desc, diff_weights_peephole_desc) in rnn_common_bwd_desc_init()
396 diff_weights_iter_desc, diff_weights_peephole_desc, in rnn_common_bwd_desc_init()
423 maybe_init_md(rd.diff_weights_peephole_desc, diff_weights_peephole_desc); in rnn_common_bwd_desc_init()
633 const dnnl_memory_desc_t *diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v2() argument
643 diff_weights_iter_desc, diff_weights_peephole_desc, nullptr, in dnnl_lstm_backward_desc_init_v2()
666 const dnnl_memory_desc_t *diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v3() argument
678 diff_weights_iter_desc, diff_weights_peephole_desc, in dnnl_lstm_backward_desc_init_v3()
H A Drnn_pd.hpp429 , diff_weights_peephole_md_(desc_.diff_weights_peephole_desc) in rnn_bwd_pd_t()
H A Dtype_helpers.hpp536 && COMPARE_DESC_MEMBERS(diff_weights_peephole_desc) in operator ==()
H A Dprimitive_hashing.cpp600 seed = hash_combine(seed, get_md_hash(desc.diff_weights_peephole_desc)); in get_desc_hash()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/include/oneapi/dnnl/
H A Ddnnl.h2881 const dnnl_memory_desc_t *diff_weights_peephole_desc,
2978 const dnnl_memory_desc_t *diff_weights_peephole_desc,
H A Ddnnl.hpp7635 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
8521 const memory::desc &diff_weights_peephole_desc, in desc()
8542 &diff_weights_peephole_desc.data, in desc()
8632 const memory::desc &diff_weights_peephole_desc, in desc()
8651 &diff_weights_peephole_desc.data, in desc()
8888 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
8889 return rnn_base::diff_weights_peephole_desc(); in diff_weights_peephole_desc()
H A Ddnnl_types.h2128 dnnl_memory_desc_t diff_weights_peephole_desc; member
/dports/misc/mxnet/incubator-mxnet-1.9.0/include/mkldnn/oneapi/dnnl/
H A Ddnnl.h2881 const dnnl_memory_desc_t *diff_weights_peephole_desc,
2978 const dnnl_memory_desc_t *diff_weights_peephole_desc,
H A Ddnnl.hpp7635 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
8521 const memory::desc &diff_weights_peephole_desc, in desc()
8542 &diff_weights_peephole_desc.data, in desc()
8632 const memory::desc &diff_weights_peephole_desc, in desc()
8651 &diff_weights_peephole_desc.data, in desc()
8888 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
8889 return rnn_base::diff_weights_peephole_desc(); in diff_weights_peephole_desc()
H A Ddnnl_types.h2128 dnnl_memory_desc_t diff_weights_peephole_desc; member
/dports/math/onednn/oneDNN-2.5.1/include/oneapi/dnnl/
H A Ddnnl.h2993 const dnnl_memory_desc_t *diff_weights_peephole_desc,
3090 const dnnl_memory_desc_t *diff_weights_peephole_desc,
H A Ddnnl.hpp8087 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
8973 const memory::desc &diff_weights_peephole_desc, in desc()
8994 &diff_weights_peephole_desc.data, in desc()
9084 const memory::desc &diff_weights_peephole_desc, in desc()
9103 &diff_weights_peephole_desc.data, in desc()
9340 memory::desc diff_weights_peephole_desc() const { in diff_weights_peephole_desc() function
9341 return rnn_base::diff_weights_peephole_desc(); in diff_weights_peephole_desc()
H A Ddnnl_types.h2339 dnnl_memory_desc_t diff_weights_peephole_desc; member