Home
last modified time | relevance | path

Searched refs:src_win_ptr (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpooling_fp32.c74 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling() local
75 tmp_avg = MS_ADD256_F32(tmp_avg, MS_LD256_F32(src_win_ptr)); in AvgPooling()
96 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling() local
97 tmp_avg = MS_ADDQ_F32(tmp_avg, MS_LDQ_F32(src_win_ptr)); in AvgPooling()
117 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling() local
118 tmp_avg += src_win_ptr[0]; in AvgPooling()
188 … const float *src_win_ptr = src_c_ptr + ((in_h_index + kh) * in_w + in_w_index + kw) * channel; in MaxPooling() local
189 tmp_max = MS_MAX256_F32(tmp_max, MS_LD256_F32(src_win_ptr)); in MaxPooling()
204 … const float *src_win_ptr = src_c_ptr + ((in_h_index + kh) * in_w + in_w_index + kw) * channel; in MaxPooling() local
205 tmp_max = MS_MAXQ_F32(tmp_max, MS_LDQ_F32(src_win_ptr)); in MaxPooling()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c78 … const float16_t *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPoolingFp16() local
80 tmp_avg = MS_ADDQ_F16(tmp_avg, MS_LDQ_F16(src_win_ptr)); in AvgPoolingFp16()
82 tmp_avg0 += src_win_ptr[0]; in AvgPoolingFp16()
83 tmp_avg1 += src_win_ptr[1]; in AvgPoolingFp16()
84 tmp_avg2 += src_win_ptr[2]; in AvgPoolingFp16()
85 tmp_avg3 += src_win_ptr[3]; in AvgPoolingFp16()
86 tmp_avg4 += src_win_ptr[4]; in AvgPoolingFp16()
87 tmp_avg5 += src_win_ptr[5]; in AvgPoolingFp16()
88 tmp_avg6 += src_win_ptr[6]; in AvgPoolingFp16()
89 tmp_avg7 += src_win_ptr[7]; in AvgPoolingFp16()
[all …]