Home
last modified time | relevance | path

Searched defs:dhc_id (Results 1 – 4 of 4) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/
H A Dref_postgemm_lstm.cpp52 const auto bias = [&](int gate_id, int dhc_id) { in lstm_fwd_postgemm_template()
75 const auto src_iter_c = [&](int mb_id, int dhc_id) { in lstm_fwd_postgemm_template()
79 const auto dst_iter_c_assign = [&](int mb_id, int dhc_id, float c_state) { in lstm_fwd_postgemm_template()
301 const auto src_iter_c = [&](int mb_id, int dhc_id) { in lstm_bwd_postgemm_template()
308 const auto dst_iter_c = [&](int mb_id, int dhc_id) { in lstm_bwd_postgemm_template()
H A Dref_postgemm_gru_lbr.cpp58 const auto bias = [&](int gate_id, int dhc_id) { in gru_lbr_fwd_postgemm_template()
H A Dref_postgemm_gru.cpp50 const auto bias = [&](int gate_id, int dhc_id) { in gru_fwd_part1_postgemm_template()
101 const auto bias = [&](int gate_id, int dhc_id) { in gru_fwd_part2_postgemm_template()
H A Dref_postgemm_rnn.cpp86 const auto bias = [&](int gate_id, int dhc_id) { in rnn_fwd_postgemm_template()