Searched refs:dst_c_ptr (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 69 float *dst_c_ptr = dst_plane_ptr + ci; in AvgPooling() local
85 MS_ST256_F32(dst_c_ptr, tmp_avg); in AvgPooling()
91 float *dst_c_ptr = dst_plane_ptr + ci; in AvgPooling() local
107 MS_STQ_F32(dst_c_ptr, tmp_avg); in AvgPooling()
112 float *dst_c_ptr = dst_plane_ptr + ci; in AvgPooling() local
128 dst_c_ptr[0] = tmp_avg; in AvgPooling()
184 float *dst_c_ptr = dst_plane_ptr + ci; in MaxPooling() local
194 MS_ST256_F32(dst_c_ptr, tmp_max); in MaxPooling()
200 float *dst_c_ptr = dst_plane_ptr + ci; in MaxPooling() local
210 MS_STQ_F32(dst_c_ptr, tmp_max); in MaxPooling()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 62 float16_t *dst_c_ptr = dst_plane_ptr + ci * C8NUM; in AvgPoolingFp16() local
99 MS_STQ_F16(dst_c_ptr, MS_MINQ_F16(MS_MAXQ_F16(tmp_avg, min_value), max_value)); in AvgPoolingFp16()
101 dst_c_ptr[0] = MSMIN(MSMAX(tmp_avg0 / (float16_t)real_count, min), max); in AvgPoolingFp16()
102 dst_c_ptr[1] = MSMIN(MSMAX(tmp_avg1 / (float16_t)real_count, min), max); in AvgPoolingFp16()
103 dst_c_ptr[2] = MSMIN(MSMAX(tmp_avg2 / (float16_t)real_count, min), max); in AvgPoolingFp16()
104 dst_c_ptr[3] = MSMIN(MSMAX(tmp_avg3 / (float16_t)real_count, min), max); in AvgPoolingFp16()
105 dst_c_ptr[4] = MSMIN(MSMAX(tmp_avg4 / (float16_t)real_count, min), max); in AvgPoolingFp16()
106 dst_c_ptr[5] = MSMIN(MSMAX(tmp_avg5 / (float16_t)real_count, min), max); in AvgPoolingFp16()
107 dst_c_ptr[6] = MSMIN(MSMAX(tmp_avg6 / (float16_t)real_count, min), max); in AvgPoolingFp16()
108 dst_c_ptr[7] = MSMIN(MSMAX(tmp_avg7 / (float16_t)real_count, min), max); in AvgPoolingFp16()
[all …]
|