Home
last modified time | relevance | path

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

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dnchw_pooling.cpp615 float *diff_dst_fp32 in execute_backward() local
622 cvt_bfloat16_to_float(diff_dst_fp32, in execute_backward()
633 = &diff_dst_fp32[diff_dst_offset + ow]; in execute_backward()
649 float *diff_dst_fp32 in execute_backward() local
658 cvt_bfloat16_to_float(diff_dst_fp32, in execute_backward()
668 = &diff_dst_fp32[diff_dst_offset + ow]; in execute_backward()
H A Dnhwc_pooling.cpp682 float *diff_dst_fp32 = &bf16cvt_ddst[ithr * OC]; in execute_backward() local
721 diff_dst_fp32, &diff_dst[dst_offset_init], OC); in execute_backward()
745 ? diff_dst_fp32[oc] in execute_backward()
774 += diff_dst_fp32[oc] / num_summands; in execute_backward()
777 = diff_dst_fp32[oc] / num_summands; in execute_backward()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dnchw_pooling.cpp625 float *diff_dst_fp32 in execute_backward() local
632 cvt_bfloat16_to_float(diff_dst_fp32, in execute_backward()
643 = &diff_dst_fp32[diff_dst_offset + ow]; in execute_backward()
659 float *diff_dst_fp32 in execute_backward() local
668 cvt_bfloat16_to_float(diff_dst_fp32, in execute_backward()
678 = &diff_dst_fp32[diff_dst_offset + ow]; in execute_backward()
H A Dnhwc_pooling.cpp684 float *diff_dst_fp32 = &bf16cvt_ddst[ithr * OC]; in execute_backward() local
723 diff_dst_fp32, &diff_dst[dst_offset_init], OC); in execute_backward()
747 ? diff_dst_fp32[oc] in execute_backward()
776 += diff_dst_fp32[oc] / num_summands; in execute_backward()
779 = diff_dst_fp32[oc] / num_summands; in execute_backward()