Home
last modified time | relevance | path

Searched defs:init_code (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dmatmul_fp16_base_coder.cc75 std::string MatMulFP16BaseCoder::InitBiasData(NNaclFp32Serializer *const init_code, CoderContext *c… in InitBiasData()
90 …ulFP16BaseCoder::InitMatrixA(NNaclFp32Serializer *const code, NNaclFp32Serializer *const init_code, in InitMatrixA()
144 …ulFP16BaseCoder::InitMatrixB(NNaclFp32Serializer *const code, NNaclFp32Serializer *const init_code, in InitMatrixB()
260 NNaclFp32Serializer code, init_code; in DoCode() local
Dlstm_fp16_coder.cc30 NNaclFp32Serializer init_code; in InitInputWeightBias() local
68 NNaclFp32Serializer init_code; in InitStateWeightBias() local
114 NNaclFp32Serializer init_code; in InitProjectWeight() local
Dmatmul_dynamic_fp16_base_coder.cc97 NNaclFp32Serializer init_code; in InitMatrixB() local
151 NNaclFp32Serializer init_code; in InitBiasData() local
Dconvolution_1x1_dynamic_fp16_coder.cc164 NNaclFp32Serializer init_code; in InitWeightBias() local
200 NNaclFp32Serializer init_code; in InitTmpBuffer() local
Dlstm_mindir_dynamic_fp16_coder.cc148 NNaclFp32Serializer init_code; in InitInputWeightBias() local
188 NNaclFp32Serializer init_code; in InitStateWeightBias() local
228 NNaclFp32Serializer init_code; in InitProjectWeight() local
Dcustom_gru_fp16_coder.cc28 void CustomGruFP16Coder::InitPackMatrixB(NNaclFp32Serializer *init_code, const std::string &src, co… in InitPackMatrixB()
Dconv_depthwise_fp16_coder.cc34 NNaclFp32Serializer init_code; in InitCodeOnline() local
Dconvolution_1x1_fp16_coder.cc90 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconv_depthwise_sw_fp16_coder.cc45 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconvolution_dynamic_fp16_coder.cc89 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconvolution_fp16_coder.cc65 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconv_depthwise_3x3_fp16_coder.cc43 NNaclFp32Serializer init_code; in InitWeightBias() local
Ddeconv2d_fp16_coder.cc60 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconvolution_winograd_fp16_coder.cc102 NNaclFp32Serializer init_code; in InitCodeOnline() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dcustom_gru_fp32_coder.cc134 void CustomGruFP32Coder::InitPackMatrixB(NNaclFp32Serializer *init_code, const std::string &src, co… in InitPackMatrixB()
139 void CustomGruFP32Coder::InitBiasCode(CoderContext *const context, NNaclFp32Serializer *init_code) { in InitBiasCode()
160 void CustomGruFP32Coder::InitWeightCode(CoderContext *const context, NNaclFp32Serializer *init_code in InitWeightCode()
184 NNaclFp32Serializer code, init_code; in DoCode() local
Dlstm_fp32_coder.cc29 NNaclFp32Serializer init_code; in InitInputWeightBias() local
65 NNaclFp32Serializer init_code; in InitStateWeightBias() local
Dconvolution_depthwise_fp32_coder.cc90 NNaclFp32Serializer init_code; in InitCodeOnline() local
Ddeconv2d_fp32_coder.cc96 NNaclFp32Serializer init_code; in InitWeightBias() local
Dconvolution_fp32_coder.cc78 NNaclFp32Serializer init_code; in InitWeightBias() local
Dgather_fp32_coder.cc35 NNaclFp32Serializer init_code; in InitCodeInChange() local
Dmatmul_fp32_base_coder.cc207 NNaclFp32Serializer code, init_code; in DoCode() local
Dconvolution_winograd_fp32_coder.cc233 NNaclFp32Serializer init_code; in InitCodeOnline() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32_grad/
Dsoftmax_cross_entropy_with_logits_coder.cc59 NNaclFp32Serializer code, init_code; in DoCode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dmatmul_base_int8_coder.cc292 NNaclInt8Serializer init_code, code; in DoCode() local
Daffine_int8_coder.cc137 NNaclInt8Serializer init_code; in PrepareFullMatmulOp() local

12