/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | lstm_cell.cpp | 35 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 D | ref_rnn_fwd.cpp | 126 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 D | ref_rnn_bwd.cpp | 142 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 D | lstm_cell.cpp | 35 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 D | ref_rnn_fwd.cpp | 126 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 D | ref_rnn_bwd.cpp | 142 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 D | ref_postgemm_lstm.cpp | 64 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 D | cell_common.cpp | 106 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 D | ref_rnn.cpp | 1107 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 D | ref_postgemm_lstm.cpp | 75 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 D | cell_common.cpp | 107 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 D | ref_rnn.cpp | 1168 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 D | jit_uni_rnn_common_postgemm.hpp | 150 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 D | brgemm_cell_common_bwd.hpp | 212 const scratch_t *scratch_gates, const float *src_iter_c,
|
H A D | brgemm_cell_common_bwd.cpp | 781 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 D | jit_uni_rnn_common_postgemm.hpp | 155 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 D | brgemm_cell_common_bwd.hpp | 212 const scratch_t *scratch_gates, const void *src_iter_c,
|
H A D | brgemm_cell_common_bwd.cpp | 780 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 D | transition-to-v1.md | 183 `src_iter_c` tensor of shape `(layers, directions, batch, channels)` for cell
|