Searched refs:oc_div4_ (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_winograd_fp32.c | 460 deconv_param->oc_div4_); in DeConvWgCalWgFp32()
464 deconv_param->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp32()
466 deconv_param->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp32()
477 DeConvWgMerge(src, dst, 4, 4, DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_div4_); in DeConvWgCalWgFp32()
486 int count = deconv_param->oc_div4_ * w_size * h_size; in DeConvWgCalCommFp32()
501 DeConvWgMerge(src, dst, 4, 4, DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_div4_); in DeConvWgCalCommFp32()
558 … deconv_param->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE * C4NUM; in DeconvWg()
575 …float *tmp_buf = (float *)unit->tmp_buffer_ + task_id * deconv_param->oc_div4_ * unit->w_size_ * u… in DeconvWg()
614 DeConvWgMerge(src, dst, src_stride, dst_stride, deconv_param->oc_div4_); in DeconvWgPost()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | deconv_winograd_fp16.c | 193 deconv_param->oc_div4_); in DeConvWgCalWgFp16()
197 deconv_param->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp16()
199 deconv_param->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE); in DeConvWgCalWgFp16()
210 DeConvWgMergeFp16(src, dst, 4, 4, DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_div4_); in DeConvWgCalWgFp16()
219 int count = deconv_param->oc_div4_ * w_size * h_size; in DeConvWgCalCommFp16()
234 DeConvWgMergeFp16(src, dst, 4, 4, DECONV_WINOGRAD_DEFAULT_TILE * deconv_param->oc_div4_); in DeConvWgCalCommFp16()
397 …float16_t *tmp_buf = (float16_t *)unit->tmp_buffer_ + task_id * deconv_param->oc_div4_ * unit->w_s… in DeconvWgFp16()
433 DeConvWgMergeFp16(src, dst, src_stride, dst_stride, deconv_param->oc_div4_); in DeconvWgPostFp16()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_winograd_fp32.cc | 165 … deconv_param_->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE * C4NUM * sizeof(float)); in InitParameter()
171 …unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * un… in InitParameter()
214 deconv_param_->oc_div4_ = UP_DIV(conv_param_->output_channel_, C4NUM); in InitComputeParam()
216 deconv_param_->oc_up4_ = deconv_param_->oc_div4_ * C4NUM; in InitComputeParam()
395 deconv_param_->oc_div4_ * C4NUM; in DoDeconv()
498 …::memset(nc4hw4_output_, 0, deconv_param_->output_plane_ * deconv_param_->oc_div4_ * C4NUM * sizeo… in Run()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_winograd_fp16.cc | 179 … deconv_param_->oc_div4_ * DECONV_WINOGRAD_DEFAULT_TILE * C4NUM * sizeof(float16_t)); in InitParameter()
184 …unit.tmp_buffer_ = malloc(deconv_param_->thread_num_ * deconv_param_->oc_div4_ * unit.w_size_ * un… in InitParameter()
199 deconv_param_->oc_div4_ * C4NUM; in DoDeconv()
261 deconv_param_->oc_div4_ = UP_DIV(conv_param_->output_channel_, C4NUM); in InitComputeParam()
263 deconv_param_->oc_up4_ = deconv_param_->oc_div4_ * C4NUM; in InitComputeParam()
463 …::memset(nc4hw4_output_, 0, deconv_param_->output_plane_ * deconv_param_->oc_div4_ * C4NUM * sizeo… in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | conv_parameter.h | 124 int oc_div4_; member
|