Searched refs:h_size_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 54 unsigned int tmp_kernel_plane = unit->w_size_ * unit->h_size_; in PackDeConvWgDataFp32()
63 for (int uhi = 0; uhi < unit->h_size_; uhi++) { in PackDeConvWgDataFp32()
80 DECONV_WINOGRAD_DEFAULT_UNIT, unit->h_size_); in PackDeConvWgDataFp32()
132 … unit->h_size_, conv_param->output_channel_, conv_param->input_channel_, false); in PackDeConvWgDataFp32()
575 …f = (float *)unit->tmp_buffer_ + task_id * deconv_param->oc_div4_ * unit->w_size_ * unit->h_size_ * in DeconvWg()
578 unit->h_size_, unit->w_size_, conv_param, deconv_param); in DeconvWg()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 244 int tmp_kernel_plane = unit->w_size_ * unit->h_size_; in PackDeConvWgDataFp16()
254 for (int uhi = 0; uhi < unit->h_size_; uhi++) { in PackDeConvWgDataFp16()
274 DECONV_WINOGRAD_DEFAULT_UNIT, unit->h_size_); in PackDeConvWgDataFp16()
314 … unit->winograd_.kh_, unit->h_size_, output_channel, conv_param->input_channel_, false); in PackDeConvWgDataFp16()
398 … unit->h_size_ * DECONV_WINOGRAD_DEFAULT_TILE * C4NUM; in DeconvWgFp16()
400 unit->h_size_, unit->w_size_, conv_param, deconv_param); in DeconvWgFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 171 …ffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * unit.h_size_ * in InitParameter()
253 unit.h_size_ = h_size; in InitComputeParam()
259 unit.winograd_.k_ = unit.h_size_; in InitComputeParam()
261 unit.winograd_.o_ = DECONV_WINOGRAD_DEFAULT_UNIT + unit.h_size_ - 1; in InitComputeParam()
262 unit.winograd_.kh_ = unit.h_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 184 …ffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * unit.h_size_ * in InitParameter()
296 unit.h_size_ = h_size; in InitComputeParam()
302 unit.winograd_.k_ = unit.h_size_; in InitComputeParam()
304 unit.winograd_.o_ = DECONV_WINOGRAD_DEFAULT_UNIT + unit.h_size_ - 1; in InitComputeParam()
305 unit.winograd_.kh_ = unit.h_size_ + DECONV_WINOGRAD_DEFAULT_UNIT - 1; in InitComputeParam()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 111 int h_size_; member
|