Home
last modified time | relevance | path

Searched defs:packed_input (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dconv1x1_int8.c19 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()
Dconv_int8.c19 void ConvInt8(int8_t *input_data, int8_t *packed_input, int8_t *matmul_input, int8_t *packed_weight, in ConvInt8()
Dpack_int8.c81 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()
Dmatmul_int8.c435 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/
Ddeconvolution_fp32_wrapper.c21 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/
Dpack_tests.cc83 …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/
Dgru_fp16.c89 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
Dconv_fp16.c23 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()
Dlstm_fp16.c245 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
Dpack_fp16.c40 …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/
Dgru_fp32.c94 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
Dadder_fp32.c53 void AdderFp32(const float *input_data, float *packed_input, const float *packed_weight, const floa… in AdderFp32()
Dlstm_fp32.c232 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
Dconv_common_fp32.c29 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()
Dpack_fp32.c32 …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/
Dpack_fp16_ext.c70 …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/
Dpack_ext.c138 …ngIm2ColPackUnitFp32(const float *input_data, const ConvParameter *conv_param, float *packed_input, in RollingIm2ColPackUnitFp32()