Searched refs:key_reducer_space (Results 1 – 6 of 6) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | cpu_reducer.cpp | 325 scratchpad.book<data_t>(key_reducer_space, space_size, PAGE_4K); in init_scratchpad() 363 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr() 421 scratchpad.book<data_t>(key_reducer_space, space_size); in init_scratchpad() 453 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | cpu_reducer.cpp | 310 scratchpad.book<data_t>(key_reducer_space, space_size, PAGE_4K); in init_scratchpad() 348 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr() 406 scratchpad.book<data_t>(key_reducer_space, space_size); in init_scratchpad() 438 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | cpu_reducer.cpp | 325 scratchpad.book<data_t>(key_reducer_space, space_size, PAGE_4K); in init_scratchpad() 363 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr() 421 scratchpad.book<data_t>(key_reducer_space, space_size); in init_scratchpad() 453 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | cpu_reducer.cpp | 310 scratchpad.book<data_t>(key_reducer_space, space_size, PAGE_4K); in init_scratchpad() 348 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr() 406 scratchpad.book<data_t>(key_reducer_space, space_size); in init_scratchpad() 438 auto space = scratchpad.template get<data_t>(key_reducer_space); in get_local_ptr()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | memory_tracking.hpp | 239 key_reducer_space, enumerator
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | memory_tracking.hpp | 239 key_reducer_space, enumerator
|