| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | deconvolution_int8.cc | 200 tmp_buffer_ = reinterpret_cast<int32_t *>(ctx_->allocator->Malloc(size * sizeof(int32_t))); in InitRunBuf()
201 if (tmp_buffer_ == nullptr) { in InitRunBuf()
221 if (tmp_buffer_ != nullptr) { in FreeRunBuf()
222 ctx_->allocator->Free(tmp_buffer_); in FreeRunBuf()
223 tmp_buffer_ = nullptr; in FreeRunBuf()
263 …tmp_buffer_ + task_id * thread_stride_ * C4NUM * input_plane * kernel_plane, weight_sum_, input_su… in DoDeconv()
267 DeConvPostInt8(tmp_buffer_ + task_id * thread_stride_ * C4NUM * input_plane * kernel_plane, in DoDeconv()
|
| D | deconvolution_int8.h | 56 int32_t *tmp_buffer_ = nullptr; /* record matmul result */
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_fp16.cc | 118 tmp_buffer_ = reinterpret_cast<float16_t *>( in InitRunBuf()
120 if (tmp_buffer_ == nullptr) { in InitRunBuf()
135 if (tmp_buffer_ != nullptr) { in FreeRunBuf()
136 ctx_->allocator->Free(tmp_buffer_); in FreeRunBuf()
137 tmp_buffer_ = nullptr; in FreeRunBuf()
169 …auto tmp_buf = tmp_buffer_ + task_id * thread_stride_ * C8NUM * kernel_plane_ * matmul_param_->row… in DoDeconv()
|
| D | deconvolution_winograd_fp16.cc | 36 if (unit.tmp_buffer_ != nullptr) { in FreeResizeBuf()
37 free(unit.tmp_buffer_); in FreeResizeBuf()
38 unit.tmp_buffer_ = nullptr; in FreeResizeBuf()
178 … unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * unit.winograd_.kh_ * unit.winograd_.kw_ * in InitParameter()
180 if (unit.tmp_buffer_ == nullptr) { in InitParameter()
184 …unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * un… in InitParameter()
186 if (unit.tmp_buffer_ == nullptr) { in InitParameter()
321 unit.tmp_buffer_ = nullptr; in InitComputeParam()
|
| D | deconvolution_fp16.h | 57 float16_t *tmp_buffer_ = nullptr; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | deconv2d_fp32_coder.cc | 40 …tmp_buffer_ = reinterpret_cast<float *>(allocator_->Malloc(kNumberTypeFloat32, tmp_buffer_size_, k… in InitRunBuf()
41 MS_CHECK_PTR(tmp_buffer_); in InitRunBuf()
168 code.CodeFunction("memset", tmp_buffer_, "0", tmp_buffer_size_); in DoCode()
185 output_ptr_, tmp_buffer_, "&matmul_parameter", "&conv_parameter"); in DoCode()
|
| D | deconv2d_fp32_coder.h | 60 float *tmp_buffer_{nullptr};
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 38 if (unit.tmp_buffer_ != nullptr) { in FreeResizeBuf()
39 free(unit.tmp_buffer_); in FreeResizeBuf()
40 unit.tmp_buffer_ = nullptr; in FreeResizeBuf()
164 … unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * unit.winograd_.kh_ * unit.winograd_.kw_ * in InitParameter()
166 if (unit.tmp_buffer_ == nullptr) { in InitParameter()
171 …unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * un… in InitParameter()
173 if (unit.tmp_buffer_ == nullptr) { in InitParameter()
280 unit.tmp_buffer_ = nullptr; in InitComputeParam()
|
| D | deconvolution_fp32.cc | 139 …auto tmp_buffer = tmp_buffer_ + task_id * thread_stride_ * C8NUM * kernel_plane_ * matmul_param_->… in DoDeconv()
212 if (tmp_buffer_ != nullptr) { in FreeRunBuf()
213 ctx_->allocator->Free(tmp_buffer_); in FreeRunBuf()
214 tmp_buffer_ = nullptr; in FreeRunBuf()
230 tmp_buffer_ = reinterpret_cast<float *>( in InitRunBuf()
232 if (tmp_buffer_ == nullptr) { in InitRunBuf()
|
| D | deconvolution_fp32.h | 62 float *tmp_buffer_ = nullptr; variable
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | deconvolution_int8_coder.cc | 112 …tmp_buffer_ = reinterpret_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, tmp_buffer_size_, k… in InitRunBuf()
136 code.CodeFunction("memset", tmp_buffer_, 0, tmp_buffer_size_); in DoCode()
156 code.CodeFunction("DeConvInt8", input_ptr_, weight_ptr_, tmp_buffer_, weight_sum_, input_sum_, in DoCode()
160 …code.CodeFunction("DeConvPostInt8", tmp_buffer_, bias_data_, tmp_output_, output_tensor_, cur_oc_r… in DoCode()
|
| D | deconvolution_int8_coder.h | 44 int32_t *tmp_buffer_{nullptr};
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | deconvolution_fp32_wrapper.h | 30 float *tmp_buffer_; member
|
| D | deconvolution_fp32_wrapper.c | 65 float *tmp_buffer = args->tmp_buffer_; in DeConvFp32Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 107 void *tmp_buffer_; member
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 557 …float *tmp_buf = (float *)unit->tmp_buffer_ + task_id * unit->winograd_.kh_ * unit->winograd_.kw_ * in DeconvWg()
575 …float *tmp_buf = (float *)unit->tmp_buffer_ + task_id * deconv_param->oc_div4_ * unit->w_size_ * u… in DeconvWg()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 381 …float16_t *tmp_buf = (float16_t *)unit->tmp_buffer_ + task_id * unit->winograd_.kh_ * unit->winogr… in DeconvWgFp16()
397 …float16_t *tmp_buf = (float16_t *)unit->tmp_buffer_ + task_id * deconv_param->oc_div4_ * unit->w_s… in DeconvWgFp16()
|