/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | lstm_cell.cpp | 37 AOC<float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_fwd_postgemm_template() local 62 dst_iter_c(ib, ih) = tmp; in lstm_fwd_postgemm_template() 149 AOC<const float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_bwd_pregemm_template() local 168 float tanhC = func1(prb.linear_cscale, dst_iter_c(ib, ih)); in lstm_bwd_pregemm_template() 220 AOC<const float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_bwd_weights_peephole() local 227 += b_gates(ib, LSTM_O, ih) * dst_iter_c(ib, ih); in lstm_bwd_weights_peephole()
|
H A D | ref_rnn_fwd.cpp | 163 AOC<float> dst_iter_c( in copy_res_fwd() local 205 &dst_iter_c(lay, dir_val, 0, 0)); in copy_res_fwd() 222 float *dst_iter_c, float *gates, float *ht, const float *weights_layer, in rnn_cell_fwd() argument 240 lstm_fwd(prb, dst_layer, dst_iter, dst_iter_c, gates, ht, in rnn_cell_fwd()
|
H A D | ref_rnn_bwd.cpp | 145 const float *bias, const float *dst_layer, const float *dst_iter_c, in rnn_cell_bwd() argument 157 dst_layer, dst_iter_c, gates, ht, diff_dst_layer, in rnn_cell_bwd()
|
H A D | rnn.cpp | 990 COMPARE_DAT(DST_ITER_C, dst_iter_c, tag::abx /*ldnc*/); in doit() 1148 COMPARE_DAT(DST_ITER_C, dst_iter_c, tag::abx /*ldnc*/); in doit()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | lstm_cell.cpp | 37 AOC<float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_fwd_postgemm_template() local 62 dst_iter_c(ib, ih) = tmp; in lstm_fwd_postgemm_template() 149 AOC<const float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_bwd_pregemm_template() local 168 float tanhC = func1(prb.linear_cscale, dst_iter_c(ib, ih)); in lstm_bwd_pregemm_template() 220 AOC<const float> dst_iter_c(dst_iter_c_, prb.mb, prb.wc); in lstm_bwd_weights_peephole() local 227 += b_gates(ib, LSTM_O, ih) * dst_iter_c(ib, ih); in lstm_bwd_weights_peephole()
|
H A D | ref_rnn_fwd.cpp | 163 AOC<float> dst_iter_c( in copy_res_fwd() local 205 &dst_iter_c(lay, dir_val, 0, 0)); in copy_res_fwd() 222 float *dst_iter_c, float *gates, float *ht, const float *weights_layer, in rnn_cell_fwd() argument 240 lstm_fwd(prb, dst_layer, dst_iter, dst_iter_c, gates, ht, in rnn_cell_fwd()
|
H A D | ref_rnn_bwd.cpp | 145 const float *bias, const float *dst_layer, const float *dst_iter_c, in rnn_cell_bwd() argument 157 dst_layer, dst_iter_c, gates, ht, diff_dst_layer, in rnn_cell_bwd()
|
H A D | rnn.cpp | 997 COMPARE_DAT(DST_ITER_C, dst_iter_c, tag::abx /*ldnc*/); in doit() 1155 COMPARE_DAT(DST_ITER_C, dst_iter_c, tag::abx /*ldnc*/); in doit()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
H A D | ref_postgemm_lstm.cpp | 63 ws_states_iter_c_aoc<float> dst_iter_c(rnn, dst_iter_c_, dst_iter_c_ld); in lstm_fwd_postgemm_template() local 91 dst_iter_c(i, j) = c_state; in lstm_fwd_postgemm_template() 272 ws_states_iter_aoc<float> dst_iter_c(rnn, dst_iter_c_, dst_iter_c_ld); in lstm_bwd_postgemm_template() local 285 float Ct = dst_iter_c(i, j); in lstm_bwd_postgemm_template()
|
H A D | cell_common.cpp | 104 ws_states_iter_c_aoc<const float> dst_iter_c( in lstm_bwd_weights_peephole_and_bias() local 123 auto &c_states = g < 2 ? src_iter_c : dst_iter_c; in lstm_bwd_weights_peephole_and_bias()
|
H A D | ref_rnn.cpp | 1123 auto dst_iter_c = rnn.is_fwd ? CTX_OUT_MEM(float *, DNNL_ARG_DST_ITER_C) in execute_() local 1286 (dst_iter_t *)dst_iter, dst_iter_c, ws_states_layer, ws_states_iter, in execute_() 1313 copy_res_iter(rnn, (float *)dst_iter, dst_iter_c, diff_src_iter, in execute_() 1318 copy_res_iter(rnn, (dst_iter_t *)dst_iter, dst_iter_c, in execute_()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
H A D | ref_postgemm_lstm.cpp | 68 const auto dst_iter_c = rnn_utils::make_raw_aoc(dst_iter_c_, in lstm_fwd_postgemm_template() local 81 = const_cast<void *>(dst_iter_c(mb_id, dhc_id)); in lstm_fwd_postgemm_template() 308 const auto dst_iter_c = [&](int mb_id, int dhc_id) { in lstm_bwd_postgemm_template() local 325 const float Ct = dst_iter_c(i, j); in lstm_bwd_postgemm_template()
|
H A D | cell_common.cpp | 104 const auto dst_iter_c = rnn_utils::make_raw_aoc(dst_iter_c_, in lstm_bwd_weights_peephole_and_bias() local 126 auto &c_states = g < 2 ? src_iter_c : dst_iter_c; in lstm_bwd_weights_peephole_and_bias()
|
H A D | ref_rnn.cpp | 1184 auto dst_iter_c = CTX_OUT_MEM(void *, DNNL_ARG_DST_ITER_C); in execute_() local 1325 (dst_iter_t *)dst_iter, dst_iter_c, ws_states_layer, ws_states_iter, in execute_() 1352 copy_res_iter(rnn, (float *)dst_iter, dst_iter_c, diff_src_iter, in execute_() 1357 copy_res_iter(rnn, (dst_iter_t *)dst_iter, dst_iter_c, in execute_()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
H A D | jit_uni_rnn_common_postgemm.hpp | 148 rnn_utils::ws_states_iter_c_aoc<float> dst_iter_c( in postgemm_fwd_call() local 174 param7_ = SAFE_PTR(dst_iter_c, m, 0); in postgemm_fwd_call() 222 rnn_utils::ws_states_iter_c_aoc<float> dst_iter_c( in rnn_postgemm_sig() local 253 param8_ = SAFE_PTR(dst_iter_c, i, 0); in rnn_postgemm_sig()
|
H A D | brgemm_cell_common_bwd.hpp | 213 const float *dst_iter_c, float *diff_weights_peephole);
|
H A D | brgemm_cell_common_bwd.cpp | 782 const float *dst_iter_c, float *diff_weights_peephole) in brgemm_diff_wei_peep_t() argument 786 , dst_iter_c_(dst_iter_c) in brgemm_diff_wei_peep_t() 813 const rnn_utils::ws_states_iter_c_aoc<const float> dst_iter_c( in kernel() local 824 const auto &c_states = g < 2 ? src_iter_c : dst_iter_c; in kernel()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
H A D | jit_uni_rnn_common_postgemm.hpp | 152 const auto dst_iter_c = rnn_utils::make_raw_aoc(dst_iter_c_, in postgemm_fwd_call() local 180 param7_ = const_cast<void *>(dst_iter_c(m, 0)); in postgemm_fwd_call() 229 const auto dst_iter_c = rnn_utils::make_raw_aoc(dst_iter_c_, in rnn_postgemm_sig() local 261 param8_ = const_cast<void *>(dst_iter_c(i, 0)); in rnn_postgemm_sig()
|
H A D | brgemm_cell_common_bwd.hpp | 213 const void *dst_iter_c, float *diff_weights_peephole);
|
H A D | brgemm_cell_common_bwd.cpp | 781 const void *dst_iter_c, float *diff_weights_peephole) in brgemm_diff_wei_peep_t() argument 785 , dst_iter_c_(dst_iter_c) in brgemm_diff_wei_peep_t() 812 const auto dst_iter_c = rnn_utils::make_raw_aoc(dst_iter_c_, in kernel() local 826 const auto &c_states = g < 2 ? src_iter_c : dst_iter_c; in kernel()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/advanced/ |
H A D | transition-to-v1.md | 186 into `dst_iter` and `dst_iter_c` respectively.
|