Home
last modified time | relevance | path

Searched refs:cell_ws_iter_offset (Results 1 – 10 of 10) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/
H A Dcell_gru_lbr.cpp39 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 Dcell_common.cpp34 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 Dcell_gru.cpp34 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 Drnn_utils.cpp390 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 Drnn_utils.hpp221 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 Dcell_gru_lbr.cpp40 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 Dcell_common.cpp35 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 Dcell_gru.cpp34 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 Drnn_utils.hpp199 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 Drnn_utils.cpp403 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()