Searched refs:kh_s (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pooling_grad.c | 44 int kh_s = MSMAX(0, over_h); in AvgPoolingGrad() local
60 for (int kh = kh_s; kh < kh_e; kh++) { in AvgPoolingGrad()
81 for (int kh = kh_s; kh < kh_e; kh++) { in AvgPoolingGrad()
122 int kh_s = MSMAX(0, over_h); in MaxPoolingGrad() local
140 for (int kh = kh_s; kh < kh_e; kh++) { in MaxPoolingGrad()
170 for (int kh = kh_s; kh < kh_e; kh++) { in MaxPoolingGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pooling_grad.c | 44 int kh_s = MSMAX(0, over_h); in AvgPoolingFp16Grad() local
60 for (int kh = kh_s; kh < kh_e; kh++) { in AvgPoolingFp16Grad()
81 for (int kh = kh_s; kh < kh_e; kh++) { in AvgPoolingFp16Grad()
125 int kh_s = MSMAX(0, over_h); in MaxPoolingFp16Grad() local
143 for (int kh = kh_s; kh < kh_e; kh++) { in MaxPoolingFp16Grad()
172 for (int kh = kh_s; kh < kh_e; kh++) { in MaxPoolingFp16Grad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 114 int kh_s = MSMAX(0, -in_h_index); in AvgPoolingOptInt8() local
116 int real_count = (kw_e - kw_s) * (kh_e - kh_s); in AvgPoolingOptInt8()
137 for (int h = kh_s; h < kh_e; h++) { in AvgPoolingOptInt8()
209 for (int h = kh_s; h < kh_e; h++) { in AvgPoolingOptInt8()
259 for (int h = kh_s; h < kh_e; h++) { in AvgPoolingOptInt8()
|
| D | pack_int8.c | 764 int kh_s = MSMAX(0, UP_DIV(-input_h, dilation_h)); in Im2ColPackUnitInt8Opt() local
769 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitInt8Opt()
776 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitInt8Opt()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pack_fp16.c | 62 int kh_s = MSMAX(0, UP_DIV(-input_h, dilation_h)); in Im2ColPackUnitFp16() local
67 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitFp16()
75 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.c | 51 int kh_s = MSMAX(0, UP_DIV(-input_h, dilation_h)); in Im2ColPackUnitFp32() local
56 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitFp32()
64 for (int j = kh_s; j < kh_e; j++) { in Im2ColPackUnitFp32()
|