Home
last modified time | relevance | path

Searched defs:wei_base (Results 1 – 16 of 16) sorted by relevance

/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Dacl_inner_product.cpp35 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
H A Dacl_convolution_utils.hpp84 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 Dacl_matmul.cpp31 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 Dacl_winograd_convolution.cpp41 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
H A Dacl_indirect_gemm_convolution.cpp28 auto wei_base = CTX_IN_MEM(const data_t *, DNNL_ARG_WEIGHTS); in execute_forward() local
H A Dacl_gemm_convolution.cpp42 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 Dgemm_x8s8s32x_convolution.cpp114 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 Dgemm_convolution.cpp50 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 Dgemm_x8s8s32x_convolution.cpp115 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 Dgemm_convolution.cpp50 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 Dgemm_bf16_convolution.cpp372 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 Djit_brgemm_1x1_conv.cpp239 const auto wei_base = jcp.wei_plain in exec_ker() local
H A Djit_brgemm_conv.cpp821 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 Dgemm_bf16_convolution.cpp362 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 Djit_brgemm_conv.cpp1034 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 Djit_brgemm_1x1_conv.cpp351 const auto wei_base = weights + wei_dt_size * wei_offset; in exec_ker() local