Searched refs:kw_e (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pooling_grad.c | 49 int kw_e = MSMIN(win_w, in_w + over_w); in AvgPoolingGrad() local
62 for (int kw = kw_s; kw < kw_e; kw++) { in AvgPoolingGrad()
83 for (int kw = kw_s; kw < kw_e; kw++) { in AvgPoolingGrad()
127 int kw_e = MSMIN(win_w, in_w + over_w); in MaxPoolingGrad() local
142 for (int kw = kw_s; kw < kw_e; kw++) { in MaxPoolingGrad()
172 for (int kw = kw_s; kw < kw_e; kw++) { in MaxPoolingGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pooling_grad.c | 49 int kw_e = MSMIN(win_w, in_w + over_w); in AvgPoolingFp16Grad() local
62 for (int kw = kw_s; kw < kw_e; kw++) { in AvgPoolingFp16Grad()
83 for (int kw = kw_s; kw < kw_e; kw++) { in AvgPoolingFp16Grad()
130 int kw_e = MSMIN(win_w, in_w + over_w); in MaxPoolingFp16Grad() local
145 for (int kw = kw_s; kw < kw_e; kw++) { in MaxPoolingFp16Grad()
174 for (int kw = kw_e; kw < kw_s; kw++) { in MaxPoolingFp16Grad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 113 int kw_e = MSMIN(win_w, in_w - in_w_index); in AvgPoolingOptInt8() local
116 int real_count = (kw_e - kw_s) * (kh_e - kh_s); in AvgPoolingOptInt8()
138 for (int w = kw_s; w < kw_e; w++) { in AvgPoolingOptInt8()
210 for (int w = kw_s; w < kw_e; w++) { in AvgPoolingOptInt8()
260 for (int w = kw_s; w < kw_e; w++) { in AvgPoolingOptInt8()
|
| D | pack_int8.c | 767 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitInt8Opt() local
773 …memcpy(matmul_input + input_plane_offset, input_data + input_x_stride, (kw_e - kw_s) * in_channel); in Im2ColPackUnitInt8Opt()
778 for (int k = kw_s; k < kw_e; ++k) { in Im2ColPackUnitInt8Opt()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | pack_fp16.c | 65 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitFp16() local
72 (kw_e - kw_s) * in_channel * sizeof(float16_t)); in Im2ColPackUnitFp16()
77 for (int n = kw_s; n < kw_e; n++) { in Im2ColPackUnitFp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.c | 54 int kw_e = MSMIN(kernel_w, UP_DIV(in_w - input_w, dilation_w)); in Im2ColPackUnitFp32() local
61 (kw_e - kw_s) * in_channel * sizeof(float)); in Im2ColPackUnitFp32()
66 for (int k = kw_s; k < kw_e; ++k) { in Im2ColPackUnitFp32()
|