Searched defs:scratch_cell_offset (Results 1 – 6 of 6) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/rnn/ |
H A D | rnn_utils.cpp | 108 size_t &scratch_cell_offset, size_t &scratchpad_size, in set_offsets() 169 scratch_cell_offset, ws_bias_offset; in get_scratchpad_and_workspace_sizes() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/rnn/ |
H A D | rnn_utils.cpp | 108 size_t &scratch_cell_offset, size_t &scratchpad_size, in set_offsets() 169 scratch_cell_offset, ws_bias_offset; in get_scratchpad_and_workspace_sizes() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/rnn/ |
H A D | rnn_utils.cpp | 308 size_t &scratch_cell_offset, size_t &ws_dhG1_offset, in set_offsets() 370 ws_diff_states_offset, ws_grid_comp_offset, scratch_cell_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 | 313 size_t &scratch_diff_states_offset, size_t &scratch_cell_offset, in set_offsets() 357 scratch_cell_offset, scratch_dhG1_offset, ws_bias_offset, in get_scratchpad_and_workspace_sizes() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ |
H A D | primitive_conf.hpp | 448 size_t scratch_cell_offset; member
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ |
H A D | primitive_conf.hpp | 481 size_t scratch_cell_offset; member
|