Searched refs:VANILLA_LSTM (Results 1 – 25 of 40) sorted by relevance
12
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/inputs/rnn/ |
| H A D | test_rnn_gpu | 12 --alg=VANILLA_LSTM --activation=UNDEF --batch=shapes_small
26 --alg=VANILLA_LSTM,LBR_GRU,VANILLA_GRU --activation=UNDEF --batch=shapes_small
31 --alg=VANILLA_LSTM
48 --alg=VANILLA_LSTM,LBR_GRU,VANILLA_GRU --prop=FWD_I,BWD_DW --batch=shapes_large
|
| H A D | perf_rnn_inference_lb | 5 --alg=VANILLA_LSTM
10 --alg=VANILLA_LSTM
|
| H A D | perf_rnn_inference_sb | 5 --alg=VANILLA_LSTM
10 --alg=VANILLA_LSTM
|
| H A D | perf_rnn_training | 4 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_training | 3 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_inference_sb | 3 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_inference_lb | 3 --alg=VANILLA_LSTM
|
| H A D | option_set_rnnt | 3 --alg=VANILLA_LSTM
|
| H A D | harness_lstm_bfloat16 | 6 --alg=VANILLA_LSTM
|
| H A D | harness_lstm_f32 | 6 --alg=VANILLA_LSTM
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/inputs/rnn/ |
| H A D | test_rnn_gpu | 12 --alg=VANILLA_LSTM --activation=UNDEF --batch=shapes_small
26 --alg=VANILLA_LSTM,LBR_GRU,VANILLA_GRU --activation=UNDEF --batch=shapes_small
31 --alg=VANILLA_LSTM
48 --alg=VANILLA_LSTM,LBR_GRU,VANILLA_GRU --prop=FWD_I,BWD_DW --batch=shapes_large
|
| H A D | perf_rnn_inference_sb | 5 --alg=VANILLA_LSTM
10 --alg=VANILLA_LSTM
|
| H A D | perf_rnn_inference_lb | 5 --alg=VANILLA_LSTM
10 --alg=VANILLA_LSTM
|
| H A D | option_set_rnnt | 3 --alg=VANILLA_LSTM
|
| H A D | perf_rnn_training | 4 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_training | 3 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_inference_lb | 3 --alg=VANILLA_LSTM
|
| H A D | option_set_perf_inference_sb | 3 --alg=VANILLA_LSTM
|
| H A D | harness_lstm_bfloat16 | 6 --alg=VANILLA_LSTM
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | ref_rnn_bwd.cpp | 33 bool is_lstm = prb.alg == VANILLA_LSTM; in prepare_ws_bwd()
82 if (prb.alg == VANILLA_LSTM) { in copy_init_bwd()
115 if (prb.alg == VANILLA_LSTM) { in copy_res_bwd()
151 case VANILLA_LSTM: in rnn_cell_bwd()
233 case VANILLA_LSTM: cell_scratchpad_size = prb.mb * prb.dhc; break; in rnn_linear_bwd()
|
| H A D | rnn.cpp | 780 = IMPLICATION(prb.with_peephole, prb.alg == VANILLA_LSTM); in check_known_skipped_case()
805 if (prb.alg != VANILLA_LSTM && prb.alg != VANILLA_GRU) { in check_known_skipped_case()
815 if (prb.alg != VANILLA_LSTM in check_known_skipped_case()
840 if (prb.is_int8() && prb.alg != VANILLA_LSTM) { in check_known_skipped_case()
940 if (prb.alg == VANILLA_LSTM) in doit()
958 if (prb.alg == VANILLA_LSTM) in doit()
989 if (prb.alg == VANILLA_LSTM) in doit()
1079 if (prb.alg == VANILLA_LSTM) in doit()
1102 if (prb.alg == VANILLA_LSTM) in doit()
1147 if (prb.alg == VANILLA_LSTM) in doit()
[all …]
|
| H A D | ref_rnn_fwd.cpp | 31 bool is_lstm = prb.alg == VANILLA_LSTM; in prepare_ws_fwd()
151 if (prb.alg == VANILLA_LSTM) in copy_init_fwd()
202 if (prb.alg == VANILLA_LSTM) { in copy_res_fwd()
228 if (prb.alg != VANILLA_LSTM) assert(dst_layer == dst_iter); in rnn_cell_fwd()
239 case VANILLA_LSTM: in rnn_cell_fwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | ref_rnn_bwd.cpp | 33 bool is_lstm = prb.alg == VANILLA_LSTM; in prepare_ws_bwd()
82 if (prb.alg == VANILLA_LSTM) { in copy_init_bwd()
115 if (prb.alg == VANILLA_LSTM) { in copy_res_bwd()
151 case VANILLA_LSTM: in rnn_cell_bwd()
233 case VANILLA_LSTM: cell_scratchpad_size = prb.mb * prb.dhc; break; in rnn_linear_bwd()
|
| H A D | rnn.cpp | 780 = IMPLICATION(prb.with_peephole, prb.alg == VANILLA_LSTM); in check_known_skipped_case()
805 if (prb.alg != VANILLA_LSTM && prb.alg != VANILLA_GRU) { in check_known_skipped_case()
815 if (prb.alg != VANILLA_LSTM in check_known_skipped_case()
840 if (prb.is_int8() && prb.alg != VANILLA_LSTM) { in check_known_skipped_case()
947 if (prb.alg == VANILLA_LSTM) in doit()
965 if (prb.alg == VANILLA_LSTM) in doit()
996 if (prb.alg == VANILLA_LSTM) in doit()
1086 if (prb.alg == VANILLA_LSTM) in doit()
1109 if (prb.alg == VANILLA_LSTM) in doit()
1154 if (prb.alg == VANILLA_LSTM) in doit()
[all …]
|
| H A D | ref_rnn_fwd.cpp | 31 bool is_lstm = prb.alg == VANILLA_LSTM; in prepare_ws_fwd()
151 if (prb.alg == VANILLA_LSTM) in copy_init_fwd()
202 if (prb.alg == VANILLA_LSTM) { in copy_res_fwd()
228 if (prb.alg != VANILLA_LSTM) assert(dst_layer == dst_iter); in rnn_cell_fwd()
239 case VANILLA_LSTM: in rnn_cell_fwd()
|
12