| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | gemm_x8s8s32x_convolution.cpp | 301 auto diff_src_base = CTX_OUT_MEM(diff_src_data_t *, DNNL_ARG_DIFF_SRC); in execute_backward_data() local
311 wei_base, bia_base, diff_src_base, scratchpad); in execute_backward_data()
324 diff_src_data_t *diff_src_base, in execute_backward_data_thr() argument
362 diff_src_data_t *__restrict diff_src = diff_src_base in execute_backward_data_thr()
|
| H A D | gemm_x8s8s32x_convolution.hpp | 184 const char *bia_base, diff_src_data_t *diff_src_base,
|
| H A D | gemm_convolution.hpp | 158 const data_t *bia_base, data_t *diff_src_base,
|
| H A D | gemm_convolution.cpp | 423 auto diff_src_base = CTX_OUT_MEM(data_t *, DNNL_ARG_DIFF_SRC); in execute_backward_data_nspc() local
431 diff_dst_base, wei_base, bia_base, diff_src_base, scratchpad); in execute_backward_data_nspc()
440 const data_t *wei_base, const data_t *bia_base, data_t *diff_src_base, in execute_backward_data_thr_nspc() argument
479 data_t *__restrict diff_src = diff_src_base + n * diff_src_mb_stride in execute_backward_data_thr_nspc()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | gemm_x8s8s32x_convolution.cpp | 311 auto diff_src_base = CTX_OUT_MEM(char *, DNNL_ARG_DIFF_SRC); in execute_backward_data() local
321 wei_base, bia_base, diff_src_base, scratchpad); in execute_backward_data()
331 const int8_t *wei_base, const char *bia_base, char *diff_src_base, in execute_backward_data_thr() argument
368 char *__restrict diff_src = diff_src_base in execute_backward_data_thr()
|
| H A D | gemm_x8s8s32x_convolution.hpp | 175 const char *bia_base, char *diff_src_base,
|
| H A D | gemm_convolution.hpp | 157 const data_t *bia_base, data_t *diff_src_base,
|
| H A D | gemm_convolution.cpp | 420 auto diff_src_base = CTX_OUT_MEM(data_t *, DNNL_ARG_DIFF_SRC); in execute_backward_data_nspc() local
428 diff_dst_base, wei_base, bia_base, diff_src_base, scratchpad); in execute_backward_data_nspc()
437 const data_t *wei_base, const data_t *bia_base, data_t *diff_src_base, in execute_backward_data_thr_nspc() argument
476 data_t *__restrict diff_src = diff_src_base + n * diff_src_mb_stride 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 | 685 auto diff_src_base = CTX_OUT_MEM(diff_src_data_t *, DNNL_ARG_DIFF_SRC); in execute_backward_data_nspc() local
693 ithr, nthr, diff_src_base, wei_base, diff_dst_base, scratchpad); in execute_backward_data_nspc()
703 const int nthr, diff_src_data_t *diff_src_base, in execute_backward_data_thr_nspc() argument
741 diff_src_data_t *__restrict diff_src = diff_src_base in execute_backward_data_thr_nspc()
|
| H A D | gemm_bf16_convolution.hpp | 285 diff_src_data_t *diff_src_base, const wei_data_t *wei_base,
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | gemm_bf16_convolution.cpp | 677 auto diff_src_base = CTX_OUT_MEM(diff_src_data_t *, DNNL_ARG_DIFF_SRC); in execute_backward_data_nspc() local
685 ithr, nthr, diff_src_base, wei_base, diff_dst_base, scratchpad); in execute_backward_data_nspc()
695 const int nthr, diff_src_data_t *diff_src_base, in execute_backward_data_thr_nspc() argument
733 diff_src_data_t *__restrict diff_src = diff_src_base in execute_backward_data_thr_nspc()
|
| H A D | gemm_bf16_convolution.hpp | 285 diff_src_data_t *diff_src_base, const wei_data_t *wei_base,
|