/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/ |
H A D | acl_inner_product.cpp | 35 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
|
H A D | acl_convolution_utils.hpp | 84 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_forward_conv_acl() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/matmul/ |
H A D | acl_matmul.cpp | 31 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/ |
H A D | acl_winograd_convolution.cpp | 41 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
|
H A D | acl_indirect_gemm_convolution.cpp | 28 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
|
H A D | acl_gemm_convolution.cpp | 42 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | gemm_x8s8s32x_convolution.cpp | 114 auto wei_base = CTX_IN_MEM(const int8_t *, DNNL_ARG_WEIGHTS); in execute_forward() local 152 const int nthr, const char *src_base, const int8_t *wei_base, in execute_forward_thr() 309 auto wei_base = CTX_IN_MEM(const int8_t *, DNNL_ARG_WEIGHTS); in execute_backward_data() local 331 const int8_t *wei_base, const char *bia_base, char *diff_src_base, in execute_backward_data_thr()
|
H A D | gemm_convolution.cpp | 50 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward_nspc() local 69 const data_t *wei_base, const data_t *bia_base, data_t *dst_base, in execute_forward_thr_nspc() 418 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_backward_data_nspc() local 437 const data_t *wei_base, const data_t *bia_base, data_t *diff_src_base, in execute_backward_data_thr_nspc()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | gemm_x8s8s32x_convolution.cpp | 115 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local 156 const wei_data_t *wei_base, const char *bia_base, dst_data_t *dst_base, in execute_forward_thr() 299 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_backward_data() local 323 const wei_data_t *wei_base, const char *bia_base, in execute_backward_data_thr()
|
H A D | gemm_convolution.cpp | 50 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward_nspc() local 69 const data_t *wei_base, const data_t *bia_base, data_t *dst_base, in execute_forward_thr_nspc() 421 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_backward_data_nspc() local 440 const data_t *wei_base, const data_t *bia_base, data_t *diff_src_base, in execute_backward_data_thr_nspc()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | gemm_bf16_convolution.cpp | 372 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_forward_nspc() local 409 const wei_data_t *wei_base, const float *bia_base, dst_data_t *dst_base, in execute_forward_thr_nspc() 684 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_backward_data_nspc() local 704 const wei_data_t *wei_base, const diff_dst_data_t *diff_dst_base, in execute_backward_data_thr_nspc()
|
H A D | jit_brgemm_1x1_conv.cpp | 239 const auto wei_base = jcp.wei_plain in exec_ker() local
|
H A D | jit_brgemm_conv.cpp | 821 const auto wei_base = weights + g * wei_ocb_sz + ocb * wei_kd_sz; in ker_base() local 984 const auto wei_base = weights + g * wei_ocb_sz + ocb * wei_kd_sz; in ker_trans() local 1099 const wei_data_t *const __restrict wei_base in ker_vpad() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | gemm_bf16_convolution.cpp | 362 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_forward_nspc() local 399 const wei_data_t *wei_base, const float *bia_base, dst_data_t *dst_base, in execute_forward_thr_nspc() 676 auto wei_base = CTX_IN_MEM(const wei_data_t *, DNNL_ARG_WEIGHTS); in execute_backward_data_nspc() local 696 const wei_data_t *wei_base, const diff_dst_data_t *diff_dst_base, in execute_backward_data_thr_nspc()
|
H A D | jit_brgemm_conv.cpp | 1034 const auto wei_base in ker_base() local 1197 const auto wei_base in ker_trans() local 1344 const char *const __restrict wei_base in ker_vpad() local
|
H A D | jit_brgemm_1x1_conv.cpp | 351 const auto wei_base = weights + wei_dt_size * wei_offset; in exec_ker() local
|