Home
last modified time | relevance | path

Searched refs:src_iter_c (Results 1 – 19 of 19) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Dlstm_cell.cpp35 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_fwd_postgemm_template() local
44 peephole_extra_i = weights_peephole(0, ih) * src_iter_c(ib, ih); in lstm_fwd_postgemm_template()
45 peephole_extra_f = weights_peephole(1, ih) * src_iter_c(ib, ih); in lstm_fwd_postgemm_template()
60 float tmp = gates(ib, LSTM_F, ih) * src_iter_c(ib, ih) in lstm_fwd_postgemm_template()
81 src_iter_c(ib, ih), dst_layer(ib, ih)); in lstm_fwd_postgemm_template()
148 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_bwd_pregemm_template() local
180 float c_old = src_iter_c(ib, ih); in lstm_bwd_pregemm_template()
219 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_bwd_weights_peephole() local
232 diff_weights_peephole(j, ih) += b_gates(ib, j, ih) * src_iter_c(ib, ih); in lstm_bwd_weights_peephole()
H A Dref_rnn_fwd.cpp126 AOC<const float> src_iter_c( in copy_init_fwd() local
152 copy(prb.mb, prb.dhc, prb.dhc, prb.wc, &src_iter_c(lay, dir_val, 0), in copy_init_fwd()
226 const float *src_layer, const float *src_iter, const float *src_iter_c, in rnn_cell_fwd() argument
243 src_layer, src_iter, src_iter_c); in rnn_cell_fwd()
H A Dref_rnn_bwd.cpp142 const float *src_layer, const float *src_iter, const float *src_iter_c, in rnn_cell_bwd() argument
155 b_gates, src_layer, src_iter, src_iter_c, weights_layer, in rnn_cell_bwd()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/
H A Dlstm_cell.cpp35 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_fwd_postgemm_template() local
44 peephole_extra_i = weights_peephole(0, ih) * src_iter_c(ib, ih); in lstm_fwd_postgemm_template()
45 peephole_extra_f = weights_peephole(1, ih) * src_iter_c(ib, ih); in lstm_fwd_postgemm_template()
60 float tmp = gates(ib, LSTM_F, ih) * src_iter_c(ib, ih) in lstm_fwd_postgemm_template()
81 src_iter_c(ib, ih), dst_layer(ib, ih)); in lstm_fwd_postgemm_template()
148 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_bwd_pregemm_template() local
180 float c_old = src_iter_c(ib, ih); in lstm_bwd_pregemm_template()
219 AOC<const float> src_iter_c(src_iter_c_, prb.mb, prb.wc); in lstm_bwd_weights_peephole() local
232 diff_weights_peephole(j, ih) += b_gates(ib, j, ih) * src_iter_c(ib, ih); in lstm_bwd_weights_peephole()
H A Dref_rnn_fwd.cpp126 AOC<const float> src_iter_c( in copy_init_fwd() local
152 copy(prb.mb, prb.dhc, prb.dhc, prb.wc, &src_iter_c(lay, dir_val, 0), in copy_init_fwd()
226 const float *src_layer, const float *src_iter, const float *src_iter_c, in rnn_cell_fwd() argument
243 src_layer, src_iter, src_iter_c); in rnn_cell_fwd()
H A Dref_rnn_bwd.cpp142 const float *src_layer, const float *src_iter, const float *src_iter_c, in rnn_cell_bwd() argument
155 b_gates, src_layer, src_iter, src_iter_c, weights_layer, in rnn_cell_bwd()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/
H A Dref_postgemm_lstm.cpp64 ws_states_iter_c_aoc<const float> src_iter_c( in lstm_fwd_postgemm_template() local
74 gate_i_arg += weights_peephole(0, j) * src_iter_c(i, j); in lstm_fwd_postgemm_template()
79 gate_f_arg += weights_peephole(1, j) * src_iter_c(i, j); in lstm_fwd_postgemm_template()
90 float c_state = gate_f * src_iter_c(i, j) + gate_i * gate_c; in lstm_fwd_postgemm_template()
273 ws_states_iter_c_aoc<const float> src_iter_c( in lstm_bwd_postgemm_template() local
301 float dG1 = src_iter_c(i, j) * dCt * x_m_square(ws_gates(i, 1, j)); in lstm_bwd_postgemm_template()
H A Dcell_common.cpp106 ws_states_iter_c_aoc<const float> src_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.cpp1107 auto src_iter_c = CTX_IN_MEM(const float *, DNNL_ARG_SRC_ITER_C); in execute_() local
1269 (const float *)src_iter, src_iter_c, diff_dst_iter, in execute_()
1274 (const src_iter_t *)src_iter, src_iter_c, diff_dst_iter, in execute_()
1285 (const src_iter_t *)src_iter, src_iter_c, (dst_layer_t *)dst_layer, in execute_()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/
H A Dref_postgemm_lstm.cpp75 const auto src_iter_c = [&](int mb_id, int dhc_id) { in lstm_fwd_postgemm_template() local
97 gate_i_arg += weights_peephole(0, j) * src_iter_c(i, j); in lstm_fwd_postgemm_template()
102 gate_f_arg += weights_peephole(1, j) * src_iter_c(i, j); in lstm_fwd_postgemm_template()
113 const float c_state = gate_f * src_iter_c(i, j) + gate_i * gate_c; in lstm_fwd_postgemm_template()
301 const auto src_iter_c = [&](int mb_id, int dhc_id) { in lstm_bwd_postgemm_template() local
342 = src_iter_c(i, j) * dCt * x_m_square(ws_gates(i, 1, j)); in lstm_bwd_postgemm_template()
H A Dcell_common.cpp107 const auto src_iter_c = rnn_utils::make_raw_aoc(src_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.cpp1168 auto src_iter_c = CTX_IN_MEM(const void *, DNNL_ARG_SRC_ITER_C); in execute_() local
1308 ws_diff_states_iter_c, (const float *)src_iter, src_iter_c, in execute_()
1313 (const src_iter_t *)src_iter, src_iter_c, diff_dst_iter, in execute_()
1324 (const src_iter_t *)src_iter, src_iter_c, (dst_layer_t *)dst_layer, in execute_()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/
H A Djit_uni_rnn_common_postgemm.hpp150 rnn_utils::ws_states_iter_c_aoc<const float> src_iter_c( in postgemm_fwd_call() local
173 : SAFE_PTR(src_iter_c, m, 0); in postgemm_fwd_call()
224 rnn_utils::ws_states_iter_c_aoc<const float> src_iter_c( in rnn_postgemm_sig() local
252 param7_ = (float *)SAFE_PTR(src_iter_c, i, 0); in rnn_postgemm_sig()
H A Dbrgemm_cell_common_bwd.hpp212 const scratch_t *scratch_gates, const float *src_iter_c,
H A Dbrgemm_cell_common_bwd.cpp781 const scratch_t *scratch_gates, const float *src_iter_c, in brgemm_diff_wei_peep_t() argument
785 , src_iter_c_(src_iter_c) in brgemm_diff_wei_peep_t()
815 const rnn_utils::ws_states_iter_c_aoc<const float> src_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.hpp155 const auto src_iter_c = rnn_utils::make_raw_aoc(src_iter_c_, in postgemm_fwd_call() local
179 param6_ = is_projection() ? src_iter_c_ : src_iter_c(m, 0); in postgemm_fwd_call()
232 const auto src_iter_c = rnn_utils::make_raw_aoc(src_iter_c_, in rnn_postgemm_sig() local
260 param7_ = const_cast<void *>(src_iter_c(i, 0)); in rnn_postgemm_sig()
H A Dbrgemm_cell_common_bwd.hpp212 const scratch_t *scratch_gates, const void *src_iter_c,
H A Dbrgemm_cell_common_bwd.cpp780 const scratch_t *scratch_gates, const void *src_iter_c, in brgemm_diff_wei_peep_t() argument
784 , src_iter_c_(src_iter_c) in brgemm_diff_wei_peep_t()
815 const auto src_iter_c = rnn_utils::make_raw_aoc(src_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.md183 `src_iter_c` tensor of shape `(layers, directions, batch, channels)` for cell