Searched refs:remain_line (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_depthwise_fp16.c | 52 float16_t *remain_line = line + ic * lw; in ConvDw3x3RowLeftFp16() local
53 memset(remain_line, 0, 16); in ConvDw3x3RowLeftFp16()
54 memset(remain_line + 8, 0, 16); in ConvDw3x3RowLeftFp16()
55 memset(remain_line + 16, 0, 16); in ConvDw3x3RowLeftFp16()
56 memset(remain_line + 24, 0, 16); in ConvDw3x3RowLeftFp16()
61 remain_line[i] = (float16_t)0.0 - d2; in ConvDw3x3RowLeftFp16()
62 remain_line[i + 8] = d1 + d2; in ConvDw3x3RowLeftFp16()
63 remain_line[i + 16] = d2 - d1; in ConvDw3x3RowLeftFp16()
64 remain_line[i + 24] = d3 - d1; in ConvDw3x3RowLeftFp16()
87 float16_t *remain_line = line + ic * lw; in ConvDw3x3RowMiddleFp16() local
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 390 float *remain_line = line + ic * lw; in ConvDw3x3RowLeft() local
391 memset(remain_line, 0, 64); in ConvDw3x3RowLeft()
396 remain_line[i] = 0.0f - d2; in ConvDw3x3RowLeft()
397 remain_line[i + 4] = d1 + d2; in ConvDw3x3RowLeft()
398 remain_line[i + 8] = d2 - d1; in ConvDw3x3RowLeft()
399 remain_line[i + 12] = d3 - d1; in ConvDw3x3RowLeft()
422 float *remain_line = line + ic * lw; in ConvDw3x3RowMiddle() local
423 memset(remain_line, 0, 64); in ConvDw3x3RowMiddle()
429 remain_line[i] = d0 - d2; in ConvDw3x3RowMiddle()
430 remain_line[i + 4] = d1 + d2; in ConvDw3x3RowMiddle()
[all …]
|