/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() 31 void Conv1x1Int8(const int8_t *packed_input, const int8_t *packed_weight, int8_t *dst, const int32_… 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()
|
D | pack_int8.c | 81 void Conv1x1PreOptPeroc(const int8_t *src_input, int8_t *packed_input, int32_t *input_sum, size_t i… in Conv1x1PreOptPeroc() 509 void Conv1x1PreOptPert(const int8_t *src_input, int8_t *packed_input, int32_t *input_sum, size_t in… in Conv1x1PreOptPert() 739 void Im2ColPackUnitInt8Opt(const int8_t *input_data, int8_t *packed_input, int8_t *matmul_input, in… in Im2ColPackUnitInt8Opt() 812 void PackInputToC8Int8(const int8_t *input_data, int16_t *packed_input, const ConvParameter *conv_p… in PackInputToC8Int8()
|
D | matmul_int8.c | 435 void PackInput4x4AndInputSumPert(const int8_t *src_input, int8_t *packed_input, int32_t *input_sum, in PackInput4x4AndInputSumPert()
|
/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() 60 const float *packed_input = args->packed_input_; in DeConvFp32Run() local
|
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
D | pack_tests.cc | 83 …auto packed_input = reinterpret_cast<float *>(malloc(in_batch * packed_input_size * sizeof(float))… in TEST_F() local 146 …auto packed_input = reinterpret_cast<float *>(malloc(in_batch * packed_input_size * sizeof(float16… in TEST_F() local
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
D | gru_fp16.c | 89 void GruUnidirectionalFp16(float16_t *output, const float16_t *packed_input, const float16_t *weigh… in GruUnidirectionalFp16() 120 float16_t *packed_input = buffer[0]; in GruFp16() local
|
D | conv_fp16.c | 23 void ConvFp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packed_weight, in ConvFp16() 65 void ConvOutNc8hw8Fp16(const float16_t *input_data, float16_t *packed_input, const float16_t *packe… in ConvOutNc8hw8Fp16()
|
D | lstm_fp16.c | 245 void LstmUnidirectionalFp16(float16_t *output, const float16_t *packed_input, const float16_t *weig… in LstmUnidirectionalFp16() 279 float16_t *packed_input = buffer[0]; in LstmFp16() local
|
D | pack_fp16.c | 40 …PackUnitFp16(const float16_t *input_data, const ConvParameter *conv_param, float16_t *packed_input, in Im2ColPackUnitFp16()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
D | gru_fp32.c | 94 void GruUnidirectional(float *output, const float *packed_input, const float *weight_g, const float… in GruUnidirectional() 124 float *packed_input = buffer[0]; in Gru() local
|
D | adder_fp32.c | 53 void AdderFp32(const float *input_data, float *packed_input, const float *packed_weight, const floa… in AdderFp32()
|
D | lstm_fp32.c | 232 void LstmUnidirectional(float *output, const float *packed_input, const float *weight_i, const floa… in LstmUnidirectional() 265 float *packed_input = buffer[0]; in Lstm() local
|
D | conv_common_fp32.c | 29 void ConvFp32(const float *input_data, float *packed_input, const float *packed_weight, const float… in ConvFp32() 111 void ConvFp32OutNC4HW4(const float *input_data, float *packed_input, const float *packed_weight, co… in ConvFp32OutNC4HW4()
|
D | pack_fp32.c | 32 …UnitFp32(const float *input_data, const ConvParameter *conv_param, float *packed_input, int real_c… in Im2ColPackUnitFp32()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
D | pack_fp16_ext.c | 70 …PackUnitFp16(const float16_t *input_data, const ConvParameter *conv_param, float16_t *packed_input, in RollingIm2ColPackUnitFp16()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
D | pack_ext.c | 138 …ngIm2ColPackUnitFp32(const float *input_data, const ConvParameter *conv_param, float *packed_input, in RollingIm2ColPackUnitFp32()
|