Home
last modified time | relevance | path

Searched defs:is_lbr (Results 1 – 14 of 14) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn_pd.hpp143 bool is_lbr() const { return cell_kind() == dnnl_lbr_gru; } in is_lbr() function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/
H A Dref_rnn_fwd.cpp261 bool is_lbr = prb.alg == LBR_GRU; in rnn_linear_fwd() local
H A Dref_rnn_bwd.cpp193 bool is_lbr = prb.alg == LBR_GRU; in rnn_linear_bwd() local
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Drnn_pd.hpp143 bool is_lbr() const { return cell_kind() == dnnl_lbr_gru; } in is_lbr() function
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/
H A Dref_rnn_fwd.cpp261 bool is_lbr = prb.alg == LBR_GRU; in rnn_linear_fwd() local
H A Dref_rnn_bwd.cpp193 bool is_lbr = prb.alg == LBR_GRU; in rnn_linear_bwd() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/
H A Drnn_utils.hpp143 bool is_fwd, is_training, is_lbr, is_int8, is_testmode, is_vanilla_gru; member
H A Dref_rnn.cpp931 bool is_lbr = this->pd()->is_lbr(); variable
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/
H A Drnn_utils.hpp166 bool is_fwd, is_training, is_lbr, is_int8, is_testmode, is_vanilla_gru; member
H A Dref_rnn.cpp933 bool is_lbr = this->pd()->is_lbr(); variable
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/
H A Drnn_utils.hpp295 bool is_fwd = 0, is_training = 0, is_lbr = 0, is_lstm_peephole = 0, member
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/
H A Drnn_utils.hpp305 bool is_fwd = 0, is_training = 0, is_lbr = 0, is_lstm_peephole = 0, member
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/
H A Dprimitive_conf.hpp389 bool is_lbr; member
/dports/math/onednn/oneDNN-2.5.1/src/gpu/
H A Dprimitive_conf.hpp417 bool is_lbr; member