Searched refs:tmp_avg (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 70 MS_FLOAT32X8 tmp_avg = MS_MOV256_F32(0); in AvgPooling() local
75 tmp_avg = MS_ADD256_F32(tmp_avg, MS_LD256_F32(src_win_ptr)); in AvgPooling()
82 tmp_avg = MS_DIV256_F32(tmp_avg, MS_MOV256_F32(real_count)); in AvgPooling()
83 tmp_avg = MS_MAX256_F32(tmp_avg, min_value_8); in AvgPooling()
84 tmp_avg = MS_MIN256_F32(tmp_avg, max_value_8); in AvgPooling()
85 MS_ST256_F32(dst_c_ptr, tmp_avg); in AvgPooling()
92 MS_FLOAT32X4 tmp_avg = MS_MOVQ_F32(0); in AvgPooling() local
97 tmp_avg = MS_ADDQ_F32(tmp_avg, MS_LDQ_F32(src_win_ptr)); in AvgPooling()
104 tmp_avg = MS_DIVQ_F32(tmp_avg, MS_MOVQ_F32(real_count)); in AvgPooling()
105 tmp_avg = MS_MAXQ_F32(tmp_avg, min_value); in AvgPooling()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 55 int16_t tmp_avg = 0; in AvgPoolingInt8() local
63 tmp_avg += *(input_ptr + in_offset); in AvgPoolingInt8()
71 int16_t tmp_out = round((float)tmp_avg / (float)real_count); in AvgPoolingInt8()
124 int16x8_t tmp_avg[2]; in AvgPoolingOptInt8() local
125 tmp_avg[0] = vmovq_n_s16(0); in AvgPoolingOptInt8()
126 tmp_avg[1] = vmovq_n_s16(0); in AvgPoolingOptInt8()
128 int16_t tmp_avg[16]; in AvgPoolingOptInt8()
131 tmp_avg[m] = 0; in AvgPoolingOptInt8()
146 tmp_avg[0] = vaddq_s16(tmp_avg[0], data1); in AvgPoolingOptInt8()
147 tmp_avg[1] = vaddq_s16(tmp_avg[1], data2); in AvgPoolingOptInt8()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 64 MS_FLOAT16X8 tmp_avg = MS_MOVQ_F16(0); in AvgPoolingFp16() local
80 tmp_avg = MS_ADDQ_F16(tmp_avg, MS_LDQ_F16(src_win_ptr)); in AvgPoolingFp16()
98 tmp_avg = MS_DIVQ_F16(tmp_avg, MS_MOVQ_F16((float16_t)real_count)); in AvgPoolingFp16()
99 MS_STQ_F16(dst_c_ptr, MS_MINQ_F16(MS_MAXQ_F16(tmp_avg, min_value), max_value)); in AvgPoolingFp16()
115 float16_t tmp_avg = 0; in AvgPoolingFp16() local
120 tmp_avg += src_win_ptr[0]; in AvgPoolingFp16()
127 tmp_avg = tmp_avg / (float16_t)real_count; in AvgPoolingFp16()
128 tmp_avg = fmax(tmp_avg, min); in AvgPoolingFp16()
129 tmp_avg = fmin(tmp_avg, max); in AvgPoolingFp16()
130 dst_c_ptr[0] = tmp_avg; in AvgPoolingFp16()
|