Searched defs:weights_layer (Results 1 – 6 of 6) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
H A D | ref_rnn_fwd.cpp | 88 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in prepare_bias() local 222 float *dst_iter_c, float *gates, float *ht, const float *weights_layer, in rnn_cell_fwd() 288 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in rnn_linear_fwd() local
|
H A D | ref_rnn_bwd.cpp | 143 const float *weights_layer, const float *weights_iter, in rnn_cell_bwd() 195 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in rnn_linear_bwd() local
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
H A D | ref_rnn_fwd.cpp | 88 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in prepare_bias() local 222 float *dst_iter_c, float *gates, float *ht, const float *weights_layer, in rnn_cell_fwd() 288 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in rnn_linear_fwd() local
|
H A D | ref_rnn_bwd.cpp | 143 const float *weights_layer, const float *weights_iter, in rnn_cell_bwd() 195 AOC<const float> weights_layer(weights_layer_, prb.n_layer, prb.n_dir(), in rnn_linear_bwd() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 169 AOC<weights_t *, 3> weights_layer( variable
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
H A D | ref_rnn.cpp | 170 const AOC<weights_t *, 3> weights_layer( variable
|