Searched refs:real_win_h_start (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 55 int real_win_h_start = MSMAX(0, -in_h_index); in AvgPoolingFp16() local
76 for (int h = real_win_h_start; h < real_win_h_end; h++) { in AvgPoolingFp16()
117 for (int h = real_win_h_start; h < real_win_h_end; h++) { in AvgPoolingFp16()
139 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC8Fp16() argument
156 for (int h = real_win_h_start; h < real_win_h_end; h++) { in MaxPoolingC8Fp16()
183 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC4Fp16() argument
203 for (int h = real_win_h_start; h < real_win_h_end; h++) { in MaxPoolingC4Fp16()
229 … float16_t min, float16_t max, int in_batch_offset, int out_plane_offset, int real_win_h_start, in MaxPoolingC1Fp16() argument
241 for (int h = real_win_h_start; h < real_win_h_end; h++) { in MaxPoolingC1Fp16()
286 int real_win_h_start = MSMAX(0, -in_h_index); in MaxPoolingFp16() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 61 int real_win_h_start = MSMAX(0, -in_h_index); in AvgPooling() local
72 for (int h = real_win_h_start; h < real_win_h_end; h++) { in AvgPooling()
94 for (int h = real_win_h_start; h < real_win_h_end; h++) { in AvgPooling()
115 for (int h = real_win_h_start; h < real_win_h_end; h++) { in AvgPooling()
176 int real_win_h_start = MSMAX(0, -in_h_index); in MaxPooling() local
186 for (int kh = real_win_h_start; kh < real_win_h_end; kh++) { in MaxPooling()
202 for (int kh = real_win_h_start; kh < real_win_h_end; kh++) { in MaxPooling()
217 for (int kh = real_win_h_start; kh < real_win_h_end; kh++) { in MaxPooling()
|