Searched refs:real_win_h_end (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pooling_fp16.c | 56 int real_win_h_end = MSMIN(win_h, in_h - 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()
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
156 for (int h = real_win_h_start; h < real_win_h_end; h++) { 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
203 for (int h = real_win_h_start; h < real_win_h_end; h++) { 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
241 for (int h = real_win_h_start; h < real_win_h_end; h++) { in MaxPoolingC1Fp16()
287 int real_win_h_end = MSMIN(win_h, in_h - in_h_index); in MaxPoolingFp16() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pooling_fp32.c | 62 int real_win_h_end = MSMIN(win_h, in_h - 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()
177 int real_win_h_end = MSMIN(win_h, in_h - 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()
|