Searched refs:in_w_index (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 49 int in_w_index = out_w_index * pooling_param->stride_w_ - pooling_param->pad_l_; in AvgPoolingFp16() local
57 int real_win_w_start = MSMAX(0, -in_w_index); in AvgPoolingFp16()
58 int real_win_w_end = MSMIN(win_w, in_w - in_w_index); in AvgPoolingFp16()
78 … const float16_t *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPoolingFp16()
119 … const float16_t *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPoolingFp16()
140 … int real_win_h_end, int real_win_w_start, int real_win_w_end, int in_h_index, int in_w_index) { in MaxPoolingC8Fp16() argument
158 int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingC8Fp16()
184 … int real_win_h_end, int real_win_w_start, int real_win_w_end, int in_h_index, int in_w_index) { in MaxPoolingC4Fp16() argument
205 int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingC4Fp16()
230 … int real_win_h_end, int real_win_w_start, int real_win_w_end, int in_h_index, int in_w_index) { in MaxPoolingC1Fp16() argument
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 49 int in_w_index = out_w_index * stride_w - pad_w; in AvgPoolingInt8() local
59 …if ((in_h_index + h) < 0 || (in_h_index + h) >= in_h || (in_w_index + w) < 0 || (in_w_index + w) >… in AvgPoolingInt8()
62 … int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPoolingInt8()
108 int in_w_index = out_w_index * pooling_param->stride_w_ - pooling_param->pad_l_; in AvgPoolingOptInt8() local
111 int input_stride = (in_h_index * in_w + in_w_index) * channel; in AvgPoolingOptInt8()
112 int kw_s = MSMAX(0, -in_w_index); in AvgPoolingOptInt8()
113 int kw_e = MSMIN(win_w, in_w - in_w_index); in AvgPoolingOptInt8()
304 int in_w_index = out_w_index * stride_w - pad_w; in MaxPoolingInt8() local
313 …if ((in_h_index + h) < 0 || (in_h_index + h) >= in_h || (in_w_index + w) < 0 || (in_w_index + w) >… in MaxPoolingInt8()
316 … int in_offset = in_channel_offset + ((in_h_index + h) * in_w + in_w_index + w) * channel; in MaxPoolingInt8()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 55 int in_w_index = out_w_index * pooling_param->stride_w_ - pooling_param->pad_l_; in AvgPooling() local
63 int real_win_w_start = MSMAX(0, -in_w_index); in AvgPooling()
64 int real_win_w_end = MSMIN(win_w, in_w - in_w_index); in AvgPooling()
74 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling()
96 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling()
117 … const float *src_win_ptr = src_c_ptr + ((in_h_index + h) * in_w + in_w_index + w) * channel; in AvgPooling()
170 int in_w_index = out_w_index * pooling_param->stride_w_ - pooling_param->pad_l_; in MaxPooling() local
178 int real_win_w_start = MSMAX(0, -in_w_index); in MaxPooling()
179 int real_win_w_end = MSMIN(win_w, in_w - in_w_index); in MaxPooling()
188 … const float *src_win_ptr = src_c_ptr + ((in_h_index + kh) * in_w + in_w_index + kw) * channel; in MaxPooling()
[all …]
|