| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | winograd_avx.c | 21 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2AvxUnit()
53 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluAvxUnit()
88 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Relu6AvxUnit()
126 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3AvxUnit()
162 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3ReluAvxUnit()
202 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3Relu6AvxUnit()
246 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2AvxUnit()
289 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2ReluAvxUnit()
335 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2Relu6AvxUnit()
383 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x3AvxUnit()
[all …]
|
| D | winograd_utils.c | 328 int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Unit()
390 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluUnit()
457 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Relu6Unit()
528 int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3Unit()
596 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3ReluUnit()
670 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3Relu6Unit()
749 int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2Unit()
821 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2ReluUnit()
898 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2Relu6Unit()
979 int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x3Unit()
[all …]
|
| D | winograd_transform.c | 131 int r_c = output_channel - j * C4NUM; in WinogradOutputNHWCTransform() local
143 int r_c = output_channel - j * C8NUM; in WinogradOutputNHWCTransform() local
188 int r_c = output_channel - j * C8NUM; in WinogradOutputNC4HW4Transform() local
201 int r_c = output_channel - j * C4NUM; in WinogradOutputNC4HW4Transform() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_utils_fp16.c | 463 int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2UnitFp16()
553 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluUnitFp16()
651 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Relu6UnitFp16()
757 int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3UnitFp16()
793 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3ReluUnitFp16()
833 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3Relu6UnitFp16()
877 int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2UnitFp16()
917 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2ReluUnitFp16()
960 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x2Relu6UnitFp16()
1006 int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform6x3UnitFp16()
[all …]
|
| D | winograd_transform_fp16.c | 154 int r_c = output_channel - j * C8NUM; in WinogradOutputNHWCTransformFp16() local
194 int r_c = output_channel - j * C8NUM; in WinogradOutputNC8HW8TransformFp16() local
|