| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | deconvolution_fp32_wrapper.c | 21 int DoDeconvFp32(const float *packed_input, const float *packed_weight, const float *packed_bias, f… in DoDeconvFp32() argument
40 …MatMulOpt(packed_input, packed_weight + task_id * thread_stride * C8NUM * kernel_plane * matmul_pa… in DoDeconvFp32()
45 …MatMulOpt(packed_input, packed_weight + task_id * thread_stride * C8NUM * kernel_plane * matmul_pa… in DoDeconvFp32()
61 const float *packed_weight = args->packed_weight_; in DeConvFp32Run() local
66 …DoDeconvFp32(packed_input, packed_weight, packed_bias, packed_output, output, tmp_buffer, matmul_p… in DeConvFp32Run()
|
| D | deconvolution_fp32_wrapper.h | 39 int DoDeconvFp32(const float *packed_input, const float *packed_weight, const float *packed_bias, f…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | conv1x1_int8.c | 19 void Conv1x1Int8Opt(const int8_t *packed_input, const int8_t *packed_weight, int8_t *dst, const int… in Conv1x1Int8Opt() argument
24 …matmul_func(packed_input, packed_weight, dst, row, col, deep4, conv_param->output_channel_, input_… in Conv1x1Int8Opt()
31 void Conv1x1Int8(const int8_t *packed_input, const int8_t *packed_weight, int8_t *dst, const int32_… in Conv1x1Int8() argument
35 MatmulInt8Opt(packed_input, packed_weight, dst, row, col, deep16, input_sum, bias, in Conv1x1Int8()
|
| D | conv_int8.c | 19 void ConvInt8(int8_t *input_data, int8_t *packed_input, int8_t *matmul_input, int8_t *packed_weight, in ConvInt8() argument
70 …gemm_input, packed_weight, gemm_output, real_cal_num, out_channel, unit_size, tmp_input_sum, bias_… in ConvInt8()
76 …matmul_func(gemm_input, packed_weight, gemm_output, real_cal_num, out_channel, unit_size, out_chan… in ConvInt8()
82 MatmulInt8Neon64(gemm_input, packed_weight, gemm_output, UP_ROUND(real_cal_num, C4NUM), in ConvInt8()
91 …gemm_input, packed_weight, gemm_output, real_cal_num, out_channel, unit_size, out_channel, tmp_inp… in ConvInt8()
|
| D | conv1x1_int8.h | 34 void Conv1x1Int8(const int8_t *packed_input, const int8_t *packed_weight, int8_t *dst, const int32_…
37 void Conv1x1Int8Opt(const int8_t *packed_input, const int8_t *packed_weight, int8_t *dst, const int…
|
| D | conv_int8.h | 36 void ConvInt8(int8_t *input_data, int8_t *packed_input, int8_t *matmul_input, int8_t *packed_weight,
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | conv1x1_init_int8_wrapper.c | 23 int8_t **packed_weight, int32_t **bias_data) { in Conv1x1Init() argument
24 if (packed_weight == NULL || bias_data == NULL) { in Conv1x1Init()
87 *packed_weight = packed_weight_; in Conv1x1Init()
|
| D | conv_init_int8_wrapper.c | 24 bool support_optimize, int8_t **packed_weight, int32_t **bias_data) { in ConvInit() argument
84 *packed_weight = packed_weight_; in ConvInit()
|
| D | conv1x1_init_int8_wrapper.h | 26 int8_t **packed_weight, int32_t **bias_data);
|
| D | conv_init_int8_wrapper.h | 24 bool support_optimize, int8_t **packed_weight, int32_t **bias_data);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | conv_fp16.h | 33 void ConvFp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packed_weight,
37 …OutNc8hw8Fp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packed_weight,
|
| D | conv_fp16.c | 23 void ConvFp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packed_weight, in ConvFp16() argument
59 …MatMulFp16(col_major_input, packed_weight, output_data + out_offset, bias_data, conv_param->act_ty… in ConvFp16()
65 …OutNc8hw8Fp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packed_weight, in ConvOutNc8hw8Fp16() argument
100 const float16_t *cur_weight = packed_weight; in ConvOutNc8hw8Fp16()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_common_fp32.c | 29 void ConvFp32(const float *input_data, float *packed_input, const float *packed_weight, const float… in ConvFp32() argument
91 …MatmulFloatAvxOpt(col_major_input, packed_weight, gemm_output, bias_data, (size_t)conv_param->act_… in ConvFp32()
94 …MatmulFloatSse64Opt(col_major_input, packed_weight, gemm_output, bias_data, (int)conv_param->act_t… in ConvFp32()
97 …MatmulFloatNeon32Opt12x4(col_major_input, packed_weight, gemm_output, bias_data, (int)conv_param->… in ConvFp32()
100 …MatmulFloatOpt(col_major_input, packed_weight, gemm_output, bias_data, conv_param->act_type_, deep… in ConvFp32()
103 …MatMul12x8(col_major_input, packed_weight, gemm_output, bias_data, (int)conv_param->act_type_, dee… in ConvFp32()
111 void ConvFp32OutNC4HW4(const float *input_data, float *packed_input, const float *packed_weight, co… in ConvFp32OutNC4HW4() argument
185 …CommonConv6x16Kernel(gemm_output + oc * output_hw, col_major_input, packed_weight + oc * deep, bia… in ConvFp32OutNC4HW4()
189 …MatmulFloatOpt(col_major_input, packed_weight, gemm_output, bias_data, conv_param->act_type_, deep… in ConvFp32OutNC4HW4()
334 void ConvSWFp32(const float *input_data, const float *packed_weight, const float *bias_data, float … in ConvSWFp32() argument
[all …]
|
| D | conv_common_fp32.h | 35 void ConvFp32(const float *input_data, float *packed_input, const float *packed_weight, const float…
39 void ConvFp32OutNC4HW4(const float *input_data, float *packed_input, const float *packed_weight, co…
55 void ConvSWFp32(const float *input_data, const float *packed_weight, const float *bias_data, float …
|
| D | adder_fp32.c | 53 void AdderFp32(const float *input_data, float *packed_input, const float *packed_weight, const floa… in AdderFp32() argument
88 …AdderOpt(col_major_gemm_input, packed_weight, gemm_output, bias_data, conv_param->act_type_, deep,… in AdderFp32()
|
| D | adder_fp32.h | 40 void AdderFp32(const float *input_data, float *packed_input, const float *packed_weight, const floa…
|
| D | conv_1x1_x86_fp32.c | 25 void Conv1x1SWFp32(const float *input_data, const float *packed_weight, const float *bias_data, flo… in Conv1x1SWFp32() argument
77 const float *weight = packed_weight + oc * kernel_step + ic * C8NUM * oc_block; in Conv1x1SWFp32()
|
| D | conv_1x1_x86_fp32.h | 34 void Conv1x1SWFp32(const float *input_data, const float *packed_weight, const float *bias_data, flo…
|