Searched refs:key_brgemm_primitive_buffer_a (Results 1 – 10 of 10) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
| H A D | memory_tracking.hpp | 164 key_brgemm_primitive_buffer_a, enumerator
|
| /dports/math/onednn/oneDNN-2.5.1/src/common/ |
| H A D | memory_tracking.hpp | 164 key_brgemm_primitive_buffer_a, enumerator
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_brgemm_inner_product_utils.cpp | 950 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
955 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
959 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
|
| H A D | jit_brgemm_inner_product.cpp | 87 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in execute_forward()
502 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in execute_backward_data()
910 buffer_a = scratchpad.template get<char>(key_brgemm_primitive_buffer_a); in thread_info_t()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_brgemm_inner_product_utils.cpp | 990 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
995 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
999 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
|
| H A D | jit_brgemm_inner_product.cpp | 89 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in execute_forward()
507 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in execute_backward_data()
916 buffer_a = scratchpad.template get<char>(key_brgemm_primitive_buffer_a); in thread_info_t()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/matmul/ |
| H A D | brgemm_matmul_utils.cpp | 620 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
|
| H A D | brgemm_matmul.cpp | 457 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in brg_matmul_exec_ctx_t()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/matmul/ |
| H A D | brgemm_matmul_utils.cpp | 942 scratchpad.book(key_brgemm_primitive_buffer_a, in init_scratchpad()
|
| H A D | brgemm_matmul.cpp | 610 ? scratchpad.template get<char>(key_brgemm_primitive_buffer_a) in brg_matmul_exec_ctx_t()
|