Home
last modified time | relevance | path

Searched refs:diff_src_loc (Results 1 – 4 of 4) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dgemm_x8s8s32x_convolution.cpp385 diff_src_data_t *__restrict diff_src_loc in execute_backward_data_thr() local
396 diff_src_loc[ic] = qz_a1b0<acc_data_t, diff_src_data_t>()(d); in execute_backward_data_thr()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dgemm_x8s8s32x_convolution.cpp409 char *__restrict diff_src_loc in execute_backward_data_thr() local
424 diff_src_md.data_type(), d, diff_src_loc, ic); in execute_backward_data_thr()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Dgemm_bf16_convolution.cpp767 diff_src_data_t *__restrict diff_src_loc in execute_backward_data_thr_nspc() local
771 cvt_float_to_bfloat16((bfloat16_t *)diff_src_loc, in execute_backward_data_thr_nspc()
779 diff_src_data_t *__restrict diff_src_loc in execute_backward_data_thr_nspc() local
785 diff_src_loc[ic] = acc_loc[ic]; in execute_backward_data_thr_nspc()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Dgemm_bf16_convolution.cpp759 diff_src_data_t *__restrict diff_src_loc in execute_backward_data_thr_nspc() local
763 cvt_float_to_bfloat16((bfloat16_t *)diff_src_loc, in execute_backward_data_thr_nspc()
771 diff_src_data_t *__restrict diff_src_loc in execute_backward_data_thr_nspc() local
777 diff_src_loc[ic] = acc_loc[ic]; in execute_backward_data_thr_nspc()