Home
last modified time | relevance | path

Searched refs:grid_ws_lay_offset (Results 1 – 4 of 4) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/
H A Drnn_utils.hpp202 const size_t &ws_states_offset_, size_t &grid_ws_lay_offset,
H A Drnn_utils.cpp380 const size_t &ws_states_offset_, size_t &grid_ws_lay_offset, in set_offsets_fwd_gemm() argument
390 grid_ws_lay_offset = (cl_ulong)(ws_states_offset_ in set_offsets_fwd_gemm()
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/
H A Drnn_utils.cpp367 const size_t &ws_states_offset_, size_t &grid_ws_lay_offset, in set_offsets_fwd_gemm() argument
377 grid_ws_lay_offset = (cl_ulong)(ws_states_offset_ in set_offsets_fwd_gemm()
H A Drnn_utils.hpp224 const size_t &ws_states_offset_, size_t &grid_ws_lay_offset,