| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | rnn_cell.cpp | 74 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in rnn_bwd_pregemm() local
|
| H A D | ref_rnn_bwd.cpp | 65 AOC<const float> diff_dst_iter( in copy_init_bwd() local
147 const float *diff_dst_iter, const float *diff_dst_iter_c, in rnn_cell_bwd()
|
| H A D | lbr_gru_cell.cpp | 99 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in lbr_gru_bwd_pregemm() local
|
| H A D | gru_cell.cpp | 122 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in gru_bwd_pregemm_part1() local
|
| H A D | lstm_cell.cpp | 251 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in lstm_bwd() local
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | rnn_cell.cpp | 74 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in rnn_bwd_pregemm() local
|
| H A D | ref_rnn_bwd.cpp | 65 AOC<const float> diff_dst_iter( in copy_init_bwd() local
147 const float *diff_dst_iter, const float *diff_dst_iter_c, in rnn_cell_bwd()
|
| H A D | lbr_gru_cell.cpp | 99 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in lbr_gru_bwd_pregemm() local
|
| H A D | gru_cell.cpp | 122 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in gru_bwd_pregemm_part1() local
|
| H A D | lstm_cell.cpp | 251 AOC<const float> diff_dst_iter(diff_dst_iter_, prb.mb, prb.wc); in lstm_bwd() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | ref_postgemm_gru_lbr.cpp | 149 ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter(rnn, diff_dst_iter_); in gru_lbr_bwd_postgemm_template() local
|
| H A D | ref_postgemm_gru.cpp | 316 ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter(rnn, diff_dst_iter_); in gru_bwd_part1_postgemm_template() local
354 ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter(rnn, diff_dst_iter_); in gru_bwd_part2_postgemm_template() local
|
| H A D | ref_postgemm_rnn.cpp | 170 ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter(rnn, diff_dst_iter_); in rnn_bwd_postgemm_template() local
|
| H A D | ref_postgemm_lstm.cpp | 278 ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter(rnn, diff_dst_iter_); in lstm_bwd_postgemm_template() local
|
| H A D | ref_rnn.cpp | 1129 auto diff_dst_iter = CTX_IN_MEM(const gemm_acc_t *, DNNL_ARG_DIFF_DST_ITER); in execute_() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | ref_postgemm_gru_lbr.cpp | 160 const ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter( in gru_lbr_bwd_postgemm_template() local
|
| H A D | ref_postgemm_gru.cpp | 338 const ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter( in gru_bwd_part1_postgemm_template() local
380 const ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter( in gru_bwd_part2_postgemm_template() local
|
| H A D | ref_postgemm_rnn.cpp | 174 const ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter( in rnn_bwd_postgemm_template() local
|
| H A D | ref_postgemm_lstm.cpp | 317 const ws_diff_states_iter_aoc<acc_data_t> diff_dst_iter( in lstm_bwd_postgemm_template() local
|
| H A D | ref_rnn.cpp | 1188 auto diff_dst_iter = CTX_IN_MEM(const gemm_acc_t *, DNNL_ARG_DIFF_DST_ITER); in execute_() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/rnn/ |
| H A D | jit_uni_rnn_common_postgemm.hpp | 218 rnn_utils::ws_diff_states_iter_aoc<gemm_acc_t> diff_dst_iter( in rnn_postgemm_sig() local
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/rnn/ |
| H A D | jit_uni_rnn_common_postgemm.hpp | 225 const rnn_utils::ws_diff_states_iter_aoc<gemm_acc_t> diff_dst_iter( in rnn_postgemm_sig() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
| H A D | ref_rnn.cpp | 1219 const memory_storage_t &diff_dst_iter, in copy_init_iter()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
| H A D | ref_rnn.cpp | 1241 const memory_storage_t &diff_dst_iter, in copy_init_iter()
|