Searched refs:kw_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 | 48 int kw_s = MSMAX(0, 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()
126 int kw_s = MSMAX(0, 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 | 48 int kw_s = MSMAX(0, 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()
129 int kw_s = MSMAX(0, 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 | 112 int kw_s = MSMAX(0, -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 | 766 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitInt8Opt() local
771 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitInt8Opt()
772 int input_plane_offset = (j * kernel_w + kw_s) * in_channel + i * in_channel * kernel_plane; in Im2ColPackUnitInt8Opt()
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 | 64 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitFp16() local
69 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitFp16()
70 int input_plane_offset = (j * kernel_w + kw_s) * in_channel + i * in_channel * kernel_plane; in Im2ColPackUnitFp16()
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 | 53 int kw_s = MSMAX(0, UP_DIV(-input_w, dilation_w)); in Im2ColPackUnitFp32() local
58 int input_x_stride = input_y_stride + kw_s * in_channel; in Im2ColPackUnitFp32()
59 int input_plane_offset = (j * kernel_w + kw_s) * in_channel + i * in_channel * kernel_plane; in Im2ColPackUnitFp32()
61 (kw_e - kw_s) * in_channel * sizeof(float)); in Im2ColPackUnitFp32()
66 for (int k = kw_s; k < kw_e; ++k) { in Im2ColPackUnitFp32()
|