Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c54 int out_channel_offset = out_plane_offset + j; in AvgPoolingInt8() local
75 *(output_ptr + out_channel_offset) = real_out; in AvgPoolingInt8()
135 int out_channel_offset = out_plane_offset + j * C16NUM; in AvgPoolingOptInt8() local
178 vst1_s8(output_ptr + out_channel_offset, real_out[0]); in AvgPoolingOptInt8()
182 vst1_s8(output_ptr + out_channel_offset + 8, real_out[1]); in AvgPoolingOptInt8()
190 *(output_ptr + out_channel_offset + l) = (int8_t)real_out[l]; in AvgPoolingOptInt8()
208 int out_channel_offset = out_c16_offset + j * C8NUM; in AvgPoolingOptInt8() local
238 vst1_s8(output_ptr + out_channel_offset, real_out); in AvgPoolingOptInt8()
246 *(output_ptr + out_channel_offset + l) = (int8_t)real_out[l]; in AvgPoolingOptInt8()
257 int out_channel_offset = out_c8_offset + k; in AvgPoolingOptInt8() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpooling_fp16.c150 int out_channel_offset = out_plane_offset + j * C8NUM; in MaxPoolingC8Fp16() local
171 vst1q_f16(output_ptr + out_channel_offset, tmp_max); in MaxPoolingC8Fp16()
176 *(output_ptr + out_channel_offset + l) = tmp_max[l]; in MaxPoolingC8Fp16()
197 int out_channel_offset = out_plane_offset + c4_offset + j * C4NUM; in MaxPoolingC4Fp16() local
218 vst1_f16(output_ptr + out_channel_offset, tmp_max); in MaxPoolingC4Fp16()
223 output_ptr[out_channel_offset + l] = tmp_max[l]; in MaxPoolingC4Fp16()
239 int out_channel_offset = out_plane_offset + k; in MaxPoolingC1Fp16() local
249 output_ptr[out_channel_offset] = tmp_max; in MaxPoolingC1Fp16()