Home
last modified time | relevance | path

Searched defs:scratch_gates_ (Results 1 – 16 of 16) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/
H A Dref_postgemm_gru_lbr.cpp40 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_lbr_fwd_postgemm_template()
140 scratch_data_t *scratch_gates_, const src_data_t *src_iter_, in gru_lbr_bwd_postgemm_template()
H A Dref_postgemm_gru.cpp44 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_fwd_part1_postgemm_template()
89 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_fwd_part2_postgemm_template()
307 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_bwd_part1_postgemm_template()
343 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_bwd_part2_postgemm_template()
H A Dcell_common.cpp99 const float *dst_iter_c_, const scratch_data_t *scratch_gates_, in lstm_bwd_weights_peephole_and_bias()
163 scratch_data_t *scratch_gates_, src_data_t *ws_ht_, in common_bwd_cell_exec_template()
H A Dref_postgemm_rnn.cpp78 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in rnn_fwd_postgemm_template()
166 scratch_data_t *scratch_gates_, acc_data_t *diff_dst_iter_, in rnn_bwd_postgemm_template()
H A Dcell_gru_lbr.cpp87 src_data_t *ws_grid_, scratch_data_t *scratch_gates_, in common_bwd_cell_exec_template()
H A Dref_postgemm_lstm.cpp42 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in lstm_fwd_postgemm_template()
260 src_data_t *ws_gates_, scratch_data_t *scratch_gates_, in lstm_bwd_postgemm_template()
H A Dcell_gru.cpp92 src_data_t *ws_gates_, scratch_data_t *scratch_gates_, in gru_bwd_cell_exec_template()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/
H A Dref_postgemm_gru_lbr.cpp40 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_lbr_fwd_postgemm_template()
149 scratch_data_t *scratch_gates_, const src_data_t *src_iter_, in gru_lbr_bwd_postgemm_template()
H A Dref_postgemm_gru.cpp44 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_fwd_part1_postgemm_template()
95 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_fwd_part2_postgemm_template()
327 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_bwd_part1_postgemm_template()
367 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in gru_bwd_part2_postgemm_template()
H A Dcell_common.cpp99 const void *dst_iter_c_, const scratch_data_t *scratch_gates_, in lstm_bwd_weights_peephole_and_bias()
170 scratch_data_t *scratch_gates_, src_data_t *ws_ht_, in common_bwd_cell_exec_template()
H A Dref_postgemm_rnn.cpp78 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in rnn_fwd_postgemm_template()
170 scratch_data_t *scratch_gates_, acc_data_t *diff_dst_iter_, in rnn_bwd_postgemm_template()
H A Dcell_gru_lbr.cpp87 src_data_t *ws_grid_, scratch_data_t *scratch_gates_, in common_bwd_cell_exec_template()
H A Dref_postgemm_lstm.cpp42 scratch_data_t *scratch_gates_, src_data_t *dst_layer_, in lstm_fwd_postgemm_template()
287 src_data_t *ws_gates_, scratch_data_t *scratch_gates_, in lstm_bwd_postgemm_template()
H A Dcell_gru.cpp91 src_data_t *ws_gates_, scratch_data_t *scratch_gates_, in gru_bwd_cell_exec_template()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/
H A Dbrgemm_cell_common_bwd.hpp222 const scratch_t *scratch_gates_; member in dnnl::impl::cpu::x64::brgemm_diff_wei_peep_t
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/
H A Dbrgemm_cell_common_bwd.hpp222 const scratch_t *scratch_gates_; member in dnnl::impl::cpu::x64::brgemm_diff_wei_peep_t