/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
H A D | cell_gru_lbr.cpp | 39 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 43 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 54 cell_ws_iter_offset, scratch_cell, 0, gemm_iter_fwd); 87 cell_ws_iter_offset, diff_weights_iter, cell_diff_wei_iter_off,
|
H A D | cell_common.cpp | 34 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 38 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 49 cell_ws_iter_offset, scratch_gates, cell_scratch_offset, 84 workspace, cell_ws_iter_offset, diff_weights_iter,
|
H A D | cell_gru.cpp | 34 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 39 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 57 cell_ws_iter_offset, scratch_gates, cell_scratch_offset, 104 workspace, cell_ws_iter_offset, diff_weights_iter,
|
H A D | rnn_utils.cpp | 390 const size_t &ws_states_offset_, size_t &cell_ws_iter_offset, in set_offsets_fwd_gemm() argument 409 cell_ws_iter_offset = (cl_ulong)(ws_states_offset_ in set_offsets_fwd_gemm() 423 size_t &cell_scratch_offset, size_t &cell_ws_iter_offset) { in set_gru_offsets_part2() argument 429 cell_ws_iter_offset = (cl_ulong)(ws_states_offset_ in set_gru_offsets_part2()
|
H A D | rnn_utils.hpp | 221 size_t &cell_scratch_offset, size_t &cell_ws_iter_offset); 228 const size_t &ws_states_offset_, size_t &cell_ws_iter_offset,
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
H A D | cell_gru_lbr.cpp | 40 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 44 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 55 cell_ws_iter_offset, scratch_cell, 0, gemm_iter_fwd); 86 cell_ws_iter_offset, diff_weights_iter, cell_diff_wei_iter_off,
|
H A D | cell_common.cpp | 35 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 39 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 50 cell_ws_iter_offset, scratch_gates, cell_scratch_offset, 83 workspace, cell_ws_iter_offset, diff_weights_iter,
|
H A D | cell_gru.cpp | 34 cl_ulong cell_scratch_offset, cell_ws_iter_offset, cell_ws_lay_offset, variable 39 ws_states_offset_, cell_ws_iter_offset, cell_ws_lay_offset, 57 cell_ws_iter_offset, scratch_gates, cell_scratch_offset, 100 workspace, cell_ws_iter_offset, diff_weights_iter,
|
H A D | rnn_utils.hpp | 199 size_t &cell_scratch_offset, size_t &cell_ws_iter_offset); 206 const size_t &ws_states_offset_, size_t &cell_ws_iter_offset,
|
H A D | rnn_utils.cpp | 403 const size_t &ws_states_offset_, size_t &cell_ws_iter_offset, in set_offsets_fwd_gemm() argument 422 cell_ws_iter_offset = (cl_ulong)(ws_states_offset_ in set_offsets_fwd_gemm() 436 size_t &cell_scratch_offset, size_t &cell_ws_iter_offset) { in set_gru_offsets_part2() argument 442 cell_ws_iter_offset = (cl_ulong)(ws_states_offset_ in set_gru_offsets_part2()
|