| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/rnn/ |
| H A D | ref_rnn_fwd.cpp | 71 AOC<const float> weights_projection( in prepare_projection_compensation() local
79 += weights_projection(layer, dir, dhc, dic); in prepare_projection_compensation()
224 const float *weights_projection, in rnn_cell_fwd() argument
242 weights_projection, weights_projection_compensation, bias, in rnn_cell_fwd()
281 AOC<const float> weights_projection( in rnn_linear_fwd() local
327 &weights_projection(lay - 1, dir_val, 0), in rnn_linear_fwd()
|
| H A D | ref_rnn_bwd.cpp | 144 const float *weights_peephole, const float *weights_projection, in rnn_cell_bwd() argument
156 weights_iter, weights_peephole, weights_projection, bias, in rnn_cell_bwd()
210 AOC<const float> weights_projection( in rnn_linear_bwd() local
282 &weights_projection(lay - 1, dir_val, 0), in rnn_linear_bwd()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/rnn/ |
| H A D | ref_rnn_fwd.cpp | 71 AOC<const float> weights_projection( in prepare_projection_compensation() local
79 += weights_projection(layer, dir, dhc, dic); in prepare_projection_compensation()
224 const float *weights_projection, in rnn_cell_fwd() argument
242 weights_projection, weights_projection_compensation, bias, in rnn_cell_fwd()
281 AOC<const float> weights_projection( in rnn_linear_fwd() local
327 &weights_projection(lay - 1, dir_val, 0), in rnn_linear_fwd()
|
| H A D | ref_rnn_bwd.cpp | 144 const float *weights_peephole, const float *weights_projection, in rnn_cell_bwd() argument
156 weights_iter, weights_peephole, weights_projection, bias, in rnn_cell_bwd()
210 AOC<const float> weights_projection( in rnn_linear_bwd() local
282 &weights_projection(lay - 1, dir_val, 0), in rnn_linear_bwd()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
| H A D | ref_rnn.cpp | 173 AOC<weights_t *, 2> weights_projection( variable
336 SAFE_PTR(weights_projection, lay, dir),
365 SAFE_PTR(weights_projection, lay, dir),
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
| H A D | ref_rnn.cpp | 174 const AOC<weights_t *, 2> weights_projection( variable
340 SAFE_PTR(weights_projection, lay, dir),
369 SAFE_PTR(weights_projection, lay, dir),
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/doc/primitives/ |
| H A D | rnn.md | 204 where \f$R\f$ is stored in `weights_projection`, and the other parameters are
|
| /dports/math/onednn/oneDNN-2.5.1/doc/primitives/ |
| H A D | rnn.md | 204 where \f$R\f$ is stored in `weights_projection`, and the other parameters are
|