Searched refs:xnor_md (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | rnn.cpp | 54 bool xnor_md(const memory_desc_t *a_md, const memory_desc_t *b_md) { in xnor_md() function
282 args_ok = args_ok && xnor_md(src_iter_desc, src_iter_c_desc) in rnn_common_fwd_desc_init()
283 && xnor_md(dst_iter_desc, dst_iter_c_desc); in rnn_common_fwd_desc_init()
375 args_ok = args_ok && xnor_md(src_iter_desc, src_iter_c_desc) in rnn_common_bwd_desc_init()
376 && xnor_md(dst_iter_desc, dst_iter_c_desc); in rnn_common_bwd_desc_init()
381 args_ok = args_ok && xnor_md(bias_desc, diff_bias_desc) in rnn_common_bwd_desc_init()
382 && xnor_md(weights_peephole_desc, diff_weights_peephole_desc) in rnn_common_bwd_desc_init()
384 && xnor_md(src_iter_desc, diff_src_iter_desc) in rnn_common_bwd_desc_init()
385 && xnor_md(src_iter_c_desc, diff_src_iter_c_desc) in rnn_common_bwd_desc_init()
386 && xnor_md(dst_iter_desc, diff_dst_iter_desc) in rnn_common_bwd_desc_init()
[all …]
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | rnn.cpp | 54 bool xnor_md(const memory_desc_t *a_md, const memory_desc_t *b_md) { in xnor_md() function
283 args_ok = args_ok && xnor_md(src_iter_desc, src_iter_c_desc) in rnn_common_fwd_desc_init()
284 && xnor_md(dst_iter_desc, dst_iter_c_desc); in rnn_common_fwd_desc_init()
376 args_ok = args_ok && xnor_md(src_iter_desc, src_iter_c_desc) in rnn_common_bwd_desc_init()
377 && xnor_md(dst_iter_desc, dst_iter_c_desc); in rnn_common_bwd_desc_init()
382 args_ok = args_ok && xnor_md(bias_desc, diff_bias_desc) in rnn_common_bwd_desc_init()
383 && xnor_md(weights_peephole_desc, diff_weights_peephole_desc) in rnn_common_bwd_desc_init()
385 && xnor_md(src_iter_desc, diff_src_iter_desc) in rnn_common_bwd_desc_init()
386 && xnor_md(src_iter_c_desc, diff_src_iter_c_desc) in rnn_common_bwd_desc_init()
387 && xnor_md(dst_iter_desc, diff_dst_iter_desc) in rnn_common_bwd_desc_init()
[all …]
|