Searched refs:set_gru_offsets_part2 (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
| H A D | cell_gru.cpp | 44 set_gru_offsets_part2(rnn, iter, dir, lay, src_t, wei_iter_offset_ptr,
|
| H A D | rnn_utils.hpp | 196 void set_gru_offsets_part2(const conf_t &rnn, int iter, int dir, int lay,
|
| H A D | rnn_utils.cpp | 433 void rnn_utils::set_gru_offsets_part2(const conf_t &rnn, int iter, int dir, in set_gru_offsets_part2() function in dnnl::impl::gpu::ocl::rnn_utils
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
| H A D | cell_gru.cpp | 44 set_gru_offsets_part2(rnn, iter, dir, lay, src_t, wei_iter_offset_ptr,
|
| H A D | rnn_utils.hpp | 218 void set_gru_offsets_part2(const conf_t &rnn, int iter, int dir, int lay,
|
| H A D | rnn_utils.cpp | 420 void rnn_utils::set_gru_offsets_part2(const conf_t &rnn, int iter, int dir, in set_gru_offsets_part2() function in dnnl::impl::gpu::ocl::rnn_utils
|