Home
last modified time | relevance | path

Searched refs:dst_iter_c (Results 1 – 21 of 21) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Dlstm_cell.cpp37 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 Dref_rnn_fwd.cpp163 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 Dref_rnn_bwd.cpp145 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 Drnn.cpp990 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 Dlstm_cell.cpp37 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 Dref_rnn_fwd.cpp163 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 Dref_rnn_bwd.cpp145 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 Drnn.cpp997 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 Dref_postgemm_lstm.cpp63 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 Dcell_common.cpp104 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 Dref_rnn.cpp1123 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 Dref_postgemm_lstm.cpp68 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 Dcell_common.cpp104 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 Dref_rnn.cpp1184 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 Djit_uni_rnn_common_postgemm.hpp148 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 Dbrgemm_cell_common_bwd.hpp213 const float *dst_iter_c, float *diff_weights_peephole);
H A Dbrgemm_cell_common_bwd.cpp782 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 Djit_uni_rnn_common_postgemm.hpp152 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 Dbrgemm_cell_common_bwd.hpp213 const void *dst_iter_c, float *diff_weights_peephole);
H A Dbrgemm_cell_common_bwd.cpp781 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 Dtransition-to-v1.md186 into `dst_iter` and `dst_iter_c` respectively.