Home
last modified time | relevance | path

Searched defs:tmp_weight (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dconvolution_depthwise_int8_coder.cc52 …int8_t *tmp_weight = reinterpret_cast<int8_t *>(allocator_->Malloc(kNumberTypeInt8, kOnlineSize, k… in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dconvolution_depthwise_int8.cc56 auto tmp_weight = reinterpret_cast<int8_t *>(malloc(pack_weight_size * sizeof(int8_t))); in InitWeightBias() local
Dconvolution_depthwise_3x3_int8.cc65 auto tmp_weight = reinterpret_cast<int8_t *>(malloc(pack_weight_size * sizeof(int8_t))); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Ddeconvolution_fp16.cc196 …auto tmp_weight = reinterpret_cast<float16_t *>(packed_weight_) + task_id * stride * matmul_param_… in DoDeconv() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dpack_fp32.c1154 …ackNHWCToNXHWCXFp32H1W1(int output_channel, int oc_block_num, int input_channel, float *tmp_weight, in PackNHWCToNXHWCXFp32H1W1()
1191 float *tmp_weight, const float *src) { in PackNHWCToNXHWCXFp32()
1251 float *tmp_weight, const float *src) { in SWPackNHWCToNXHWCXFp32()