Searched defs:ws_c_states_offset (Results 1 – 2 of 2) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
H A D | rnn_utils.cpp | 306 size_t &ws_states_offset, size_t &ws_c_states_offset, in set_offsets() 369 size_t ws_gates_offset, ws_states_offset, ws_c_states_offset, in get_scratchpad_and_workspace_sizes() local
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/rnn/ |
H A D | rnn_utils.cpp | 311 size_t &ws_states_offset, size_t &ws_c_states_offset, in set_offsets() 355 size_t ws_gates_offset, ws_states_offset, ws_c_states_offset, in get_scratchpad_and_workspace_sizes() local
|