Searched refs:skip_nonlinear (Results 1 – 16 of 16) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | bench_rnn.cpp | 43 for_(auto i_skip_nonlinear : s.skip_nonlinear) in check_correctness() 118 s.skip_nonlinear, def.skip_nonlinear, argv[0]) in bench()
|
H A D | rnn.hpp | 221 std::vector<bool> skip_nonlinear {false}; member 252 float beta, bool skip_nonlinear, bool trivial_strides, in prb_t() 269 , skip_nonlinear(skip_nonlinear) in prb_t() 400 bool skip_nonlinear; member
|
H A D | rnn_aux.cpp | 236 if (canonical || prb.skip_nonlinear != def.skip_nonlinear[0]) in operator <<() 237 s << "--skip-nonlinear=" << bool2str(prb.skip_nonlinear) << " "; in operator <<() 634 = (prb.skip_nonlinear || ((prb.n_layer == 1) && (prb.n_iter == 1))); in compare_dat() 756 if (skip_nonlinear) { in set_tparams()
|
H A D | gru_cell.cpp | 51 if (prb.skip_nonlinear) in gru_fwd_postgemm_part1() 84 if (prb.skip_nonlinear) in gru_fwd_postgemm_part2()
|
H A D | lstm_cell.cpp | 89 if (prb.skip_nonlinear) in lstm_fwd_postgemm() 203 if (prb.skip_nonlinear) in lstm_bwd_pregemm()
|
H A D | rnn_cell.cpp | 30 if (prb.skip_nonlinear) in activation()
|
H A D | lbr_gru_cell.cpp | 65 if (prb.skip_nonlinear) in lbr_gru_fwd_postgemm()
|
H A D | rnn.cpp | 104 if (prb.skip_nonlinear) in create_dnnl_rnn_attr() 386 bool flip_sign = prb.skip_nonlinear == false && prb.alg == VANILLA_RNN in fill_activation() 394 const bool special_case = prb.prop == dnnl_backward && prb.skip_nonlinear; in fill_src_iter_c()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | bench_rnn.cpp | 43 for_(auto i_skip_nonlinear : s.skip_nonlinear) in check_correctness() 118 s.skip_nonlinear, def.skip_nonlinear, argv[0]) in bench()
|
H A D | rnn.hpp | 221 std::vector<bool> skip_nonlinear {false}; member 252 float beta, bool skip_nonlinear, bool trivial_strides, in prb_t() 269 , skip_nonlinear(skip_nonlinear) in prb_t() 400 bool skip_nonlinear; member
|
H A D | rnn_aux.cpp | 236 if (canonical || prb.skip_nonlinear != def.skip_nonlinear[0]) in operator <<() 237 s << "--skip-nonlinear=" << bool2str(prb.skip_nonlinear) << " "; in operator <<() 634 = (prb.skip_nonlinear || ((prb.n_layer == 1) && (prb.n_iter == 1))); in compare_dat() 756 if (skip_nonlinear) { in set_tparams()
|
H A D | gru_cell.cpp | 51 if (prb.skip_nonlinear) in gru_fwd_postgemm_part1() 84 if (prb.skip_nonlinear) in gru_fwd_postgemm_part2()
|
H A D | lstm_cell.cpp | 89 if (prb.skip_nonlinear) in lstm_fwd_postgemm() 203 if (prb.skip_nonlinear) in lstm_bwd_pregemm()
|
H A D | rnn_cell.cpp | 30 if (prb.skip_nonlinear) in activation()
|
H A D | lbr_gru_cell.cpp | 65 if (prb.skip_nonlinear) in lbr_gru_fwd_postgemm()
|
H A D | rnn.cpp | 104 if (prb.skip_nonlinear) in create_dnnl_rnn_attr() 386 bool flip_sign = prb.skip_nonlinear == false && prb.alg == VANILLA_RNN in fill_activation() 394 const bool special_case = prb.prop == dnnl_backward && prb.skip_nonlinear; in fill_src_iter_c()
|