Searched refs:in_h_index (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 50 int in_h_index = out_h_index * pooling_param->stride_h_ - pooling_param->pad_u_; in AvgPoolingFp16() local
55 int real_win_h_start = MSMAX(0, -in_h_index); in AvgPoolingFp16()
56 int real_win_h_end = MSMIN(win_h, in_h - in_h_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 | 50 int in_h_index = out_h_index * stride_h - pad_h; 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()
109 int in_h_index = out_h_index * pooling_param->stride_h_ - pooling_param->pad_u_; in AvgPoolingOptInt8() local
111 int input_stride = (in_h_index * in_w + in_w_index) * channel; in AvgPoolingOptInt8()
114 int kh_s = MSMAX(0, -in_h_index); in AvgPoolingOptInt8()
115 int kh_e = MSMIN(win_h, pooling_param->input_h_ - in_h_index); in AvgPoolingOptInt8()
305 int in_h_index = out_h_index * stride_h - pad_h; 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 | 56 int in_h_index = out_h_index * pooling_param->stride_h_ - pooling_param->pad_u_; in AvgPooling() local
61 int real_win_h_start = MSMAX(0, -in_h_index); in AvgPooling()
62 int real_win_h_end = MSMIN(win_h, in_h - in_h_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()
171 int in_h_index = out_h_index * pooling_param->stride_h_ - pooling_param->pad_u_; in MaxPooling() local
176 int real_win_h_start = MSMAX(0, -in_h_index); in MaxPooling()
177 int real_win_h_end = MSMIN(win_h, in_h - in_h_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 …]
|